tests/trainer/test_trainer_seq2seq.py · 1e402b957d96597e5e47c06da5671ccec09621cc · chenpangpang / transformers

"tests/albert/test_modeling_albert.py" did not exist on "b6ea0f43aeb7ff1dcb03658e38bacae1130abd91"

Have seq2seq just use gather (#27025) · 067c4a31

Zach Mueller authored Nov 14, 2023



* Have seq2seq just use gather

* Change

* Reset after

* Make slow

* Apply suggestions from code review
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

* Clean

* Simplify and just use gather

* Update tests/trainer/test_trainer_seq2seq.py
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* gather always for seq2seq

---------
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

067c4a31

test_trainer_seq2seq.py 7.05 KB

Replace test_trainer_seq2seq.py