cyyever 41a0e58e5b Set weights_only in torch.load (#36991)

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