gautham 48385aa4f4 Added support for seed in DataCollatorForWholeWordMask (#36903) ... * Added support for seed in `DataCollatorForWholeWordMask`, and also wrote tests. Also fixed bugs where the code hardcoded values for mask replacement probability and random replacement probability, instead of using the values passed by the user. * formatting issues * Used better way to generate seed in TF. Made tests more consistent.		2025-03-24 16:57:17 +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	HPU support (#36424)	2025-03-12 09:08:12 +01:00
test_trainer_fsdp.py	HPU support (#36424)	2025-03-12 09:08:12 +01:00
test_trainer_seq2seq.py	fix typos in the tests directory (#36717)	2025-03-17 17:45:57 +00: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	chore: fix typos in the tests directory (#36813)	2025-03-21 10:20:05 +01:00