tests/trainer/test_trainer.py · c33f6046c3dab8f41bedf893404e6469dea3bce8 · 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.py 87.4 KB

Replace test_trainer.py