transformers
Fix batch size handling in prediction_loop for DataLoaderShard
#34343

Merged

Fix batch size handling in prediction_loop for DataLoaderShard #34343

ArthurZucker merged 6 commits into huggingface:main from zeus2611:trainer-dataloadershard-fix

Fix batch size handling in prediction_loop for DataLoaderShard

196645d5

Merge branch 'huggingface:main' into trainer-dataloadershard-fix

c057bf17

Merge branch 'huggingface:main' into trainer-dataloadershard-fix

e546b1b1

muellerzr requested changes on 2024-10-24

Update src/transformers/trainer.py

739ba423

muellerzr approved these changes on 2024-10-24

muellerzr requested a review from

ArthurZucker 1 year ago

Merge branch 'huggingface:main' into trainer-dataloadershard-fix

13aaa947

Applied the fix to remove unused imports

5ca40f9f

SunMarc approved these changes on 2024-10-28

ArthurZucker approved these changes on 2024-10-28

ArthurZucker merged 92bcdff2 into main 1 year ago

zeus2611 deleted the trainer-dataloadershard-fix branch 1 year ago

Reviewers

ArthurZucker

SunMarc

muellerzr

Assignees

No one assigned

Labels

None yet

Milestone

No milestone