Marc Sun bfc9ddf5c6 Add StableAdamW Optimizer (#39446) ...

* Added StableAdamW as an optimizer option for Trainer. Also wrote tests to verify its behaviour.

* Fixed issue with

* Added docs for StableAdamW. Also fixed a typo in schedule free optimizers

---------

Co-authored-by: Gautham Krithiwas <gauthamkrithiwas2003@gmail.com>

2025-07-16 13:35:53 +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

[tests] remove TF tests (uses of require_tf) (#38944)

2025-06-25 17:29:10 +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

Gaudi3 CI (#38790)

2025-06-23 10:56:51 +02:00

test_trainer_distributed_worker_seed.py

Gaudi3 CI (#38790)

2025-06-23 10:56:51 +02:00

test_trainer_distributed.py

Gaudi3 CI (#38790)

2025-06-23 10:56:51 +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

[generation] Less verbose warnings by default (#38179)

2025-05-19 10:03:37 +00: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

Add StableAdamW Optimizer (#39446)

2025-07-16 13:35:53 +02:00