transformers/tests/trainer
ivarflakstad e27d230ddd
Disable report callbacks for certain training tests (#38088)
* Disable report callbacks for certain training tests

* Disable report callbacks for test_auto_batch_size_finder
2025-05-13 14:49:55 +02:00
..
__init__.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_data_collator.py add FlashAttentionKwargs and seq_idx to flat collator (#36456) 2025-04-16 15:45:03 +02: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_worker_seed.py update seed_worker to set seed based on worker_id and rank (#37980) 2025-05-12 15:59:16 +00:00
test_trainer_distributed.py Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
test_trainer_fsdp.py fix FSDP + torch.compile bug when saving pretrained model (#37725) 2025-05-06 17:51:28 +02:00
test_trainer_seq2seq.py Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
test_trainer_tpu.py Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
test_trainer_utils.py Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
test_trainer.py Disable report callbacks for certain training tests (#38088) 2025-05-13 14:49:55 +02:00