tests/test_trainer.py · d03695f3a2c7abe9f790dc86492697dd7d9e539e · chenpangpang / transformers

[trainer] implement support for full fp16 in evaluation/predict (#10268) · 4eddc459

Stas Bekman authored Feb 18, 2021



* implement --fp16_full_eval

* Apply suggestions from code review
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* style

* add test
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

4eddc459

test_trainer.py 43.5 KB

Replace test_trainer.py