transformers
Fix batch size handling in prediction_loop for DataLoaderShard
#34343

Merged

Commits

Fix batch size handling in prediction_loop for DataLoaderShard

zeus2611 committed 1 year ago
Merge branch 'huggingface:main' into trainer-dataloadershard-fix

zeus2611 committed 1 year ago
Merge branch 'huggingface:main' into trainer-dataloadershard-fix

zeus2611 committed 1 year ago
Update src/transformers/trainer.py

zeus2611 committed 1 year ago
Merge branch 'huggingface:main' into trainer-dataloadershard-fix

zeus2611 committed 1 year ago
Applied the fix to remove unused imports

zeus2611 committed 1 year ago