src/transformers/trainer_tf.py · 3f40070c88de07169fe18b0b4c4003ef2858a284 · chenpangpang / transformers

Gradient accumulation for TFTrainer (#9585) · 3f40070c

Kiyoung Kim authored Jan 15, 2021



* gradient accumulation for tftrainer

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

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

3f40070c

trainer_tf.py 32.7 KB

Replace trainer_tf.py