tests/trainer/test_trainer.py · 636b03244cb3c5bac6d12a5a968d5024e0fde7c3 · chenpangpang / transformers

Fix trainer test wrt DeepSpeed + auto_find_bs (#29061) · 636b0324

Zach Mueller authored Feb 16, 2024



* FIx trainer test

* Update tests/trainer/test_trainer.py
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

---------
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

636b0324

test_trainer.py 137 KB

Replace test_trainer.py