src/transformers/trainer_pt_utils.py · d7633a4e468c8d720c6de9d363c5605a0bfb9ed9 · chenpangpang / transformers

Sylvain Gugger authored Apr 26, 2021

* Add FP16 support for SageMaker MP

* Add print debugs

* Squeeze

* Remove debug statements

* Add defensive check

* Typo

d7633a4e

trainer_pt_utils.py 42.8 KB