Zach Mueller 067c4a310d Have seq2seq just use gather (#27025) ... * 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>		2023-11-14 14:54:44 -05:00
..
__init__.py	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00
test_data_collator.py	handle numpy inputs in whole word mask data collator (#22032)	2023-03-10 10:50:29 -05:00
test_trainer_callback.py	Apply ruff flake8-comprehensions (#21694)	2023-02-22 09:14:54 +01:00
test_trainer_distributed.py	Patch with accelerate xpu (#25714)	2023-09-05 15:41:42 +01:00
test_trainer_seq2seq.py	Have seq2seq just use gather (#27025)	2023-11-14 14:54:44 -05:00
test_trainer_tpu.py	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00
test_trainer_utils.py	Apply ruff flake8-comprehensions (#21694)	2023-02-22 09:14:54 +01:00
test_trainer.py	enable memory tracker metrics for npu (#27280)	2023-11-06 13:44:21 +00:00