tests/test_trainer_utils.py · f284089ec4cf6da2d659512b09f2cf526dc6b308 · chenpangpang / transformers

Fix memory regression in Seq2Seq example (#9713) · 5f80c15e

Sylvain Gugger authored Jan 21, 2021

* Fix memory regression in Seq2Seq example

* Fix test and properly deal with -100

* Easier condition with device safety

* Patch for MBartTokenzierFast

5f80c15e

test_trainer_utils.py 5.09 KB

Replace test_trainer_utils.py