transformers
2fbb2379
- Add the auto_find_batch_size capability from Accelerate into Trainer (#17068)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
4 years ago
Add the auto_find_batch_size capability from Accelerate into Trainer (#17068) Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> - Adds auto_batch_size finder - Moves training loop to an inner training loop
References
#17068 - Add the auto_find_batch_size capability from Accelerate into Trainer
Author
muellerzr
Parents
df735d13
Loading