tests/trainer/test_trainer_distributed.py · 03462875cc2d6506eb66a74de7d19b93ce968596 · chenpangpang / transformers

Zachary Mueller authored Apr 17, 2023



* Use accelerate for device management

* Add accelerate to setup
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

03462875

test_trainer_distributed.py 5.58 KB