tests/test_trainer_utils.py · d9c62047a8d75e18d2849d345ab3394875a712ef · chenpangpang / transformers

Trainer support for IterableDataset for evaluation and predict (#11286) · d9c62047

Sylvain Gugger authored Apr 16, 2021

* Bulk of the work

* Polish and tests

* Update QA Trainer

* Avoid breaking the predict method

* Deprecation warnings

* Store real eval dataloder

* Get eval dataset reference before wrap

d9c62047

test_trainer_utils.py 15.6 KB

Replace test_trainer_utils.py