tests/test_trainer_tpu.py · c108d0b5a43fee12e1ef578fe871f0f123b06018 · chenpangpang / transformers

"sgl-router/git@developer.sourcefind.cn:zhaoyu6/sglang.git" did not exist on "9e169ea8b51983cbe261da92c56e5780040ed593"

Add predict step accumulation (#7767) · a1d1b332

Sylvain Gugger authored Oct 14, 2020



* Add eval_accumulation_step and clean distributed eval

* Add TPU test

* Add TPU stuff

* Fix arg name

* Fix Seq2SeqTrainer

* Fix total_size

* Update src/transformers/trainer_pt_utils.py
Co-authored-by: Lysandre Debut <lysandre@huggingface.co>

* Doc and add test to TPU

* Add unit test

* Adapt name
Co-authored-by: Lysandre Debut <lysandre@huggingface.co>

a1d1b332

test_trainer_tpu.py 3.35 KB

Replace test_trainer_tpu.py