pytorch-screen.py: fix attribute error (backend einsum) on newer pytorch

aws-samples / awsome-distributed-training

Collection of best practices, reference architectures, model training examples and utilities to train large models on AWS.

MIT No Attribution

134 stars 57 forks source link

Closed verdimrc closed 1 month ago

verdimrc commented 1 month ago

Issue #, if available: pytorch-screen.py throws attribute error on newer PyTorch.

Description of changes: fix attribute error when running pytorch-screen.py on newer PyTorch.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.