Shiyu a63cb7578e update seed_worker to set seed based on worker_id and rank (#37980) ... * update seed_worker to set seed based on worker_id and rank * test case * set output_dir as remove tmp dir		2025-05-12 15:59:16 +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	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	Fix tot update in trainer (#37923)	2025-05-12 17:45:24 +02:00