tests/test_trainer.py · 6290169eb3391d72d9a08cab5c54a54b73a87463 · chenpangpang / transformers

Sylvain Gugger authored Mar 04, 2021

* Rework TPU checkpointing in Trainer

* Wraps the barrier in a dist test

* Address review comments

* Remove line

6290169e

test_trainer.py 45.2 KB