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