transformers/tests/trainer
2025-03-27 14:55:50 +00:00
..
__init__.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_data_collator.py Added support for seed in DataCollatorForWholeWordMask (#36903) 2025-03-24 16:57:17 +00:00
test_trainer_callback.py fix: prevent second save in the end of training if last step was saved already (#36219) 2025-02-20 17:38:52 +01:00
test_trainer_distributed_loss.py Fix multi gpu loss sync condition, add doc and test (#35743) 2025-02-12 15:41:31 +01:00
test_trainer_distributed.py Remove deprecated training arguments (#36946) 2025-03-26 16:44:48 +00:00
test_trainer_fsdp.py HPU support (#36424) 2025-03-12 09:08:12 +01:00
test_trainer_seq2seq.py Update ruff to 0.11.2 (#36962) 2025-03-25 16:00:11 +01: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 layernorm_decay_fix (#35927) 2025-02-04 11:01:49 +01:00
test_trainer.py Set weights_only in torch.load (#36991) 2025-03-27 14:55:50 +00:00