tests/trainer/test_trainer_utils.py · 2fbb237967f5d1b2eb65c2131954f23a24bd29ef · chenpangpang / transformers

Add the auto_find_batch_size capability from Accelerate into Trainer (#17068) · 2fbb2379

Zachary Mueller authored May 09, 2022


Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

- Adds auto_batch_size finder 
- Moves training loop to an inner training loop

2fbb2379

test_trainer_utils.py 19.8 KB

Replace test_trainer_utils.py