Wing Lian 0542c7dd56 add muon optimizer ...

optimizer_cls_and_kwargs is on trainer_kwargs
only add adamw_kwargs if they're non-null
fix mocks
better handling of override and check the optimizer
unwrap optimizer

2025-03-05 10:47:22 -05:00

..

__init__.py

CLI Implementation with Click (#2107)

2024-12-05 22:11:48 -05:00

conftest.py

CLI cleanup and documentation (#2244)

2025-01-13 17:55:29 +00:00

test_cli_base.py

Basic evaluate CLI command / codepath (#2188)

2024-12-16 15:46:31 -05:00

test_cli_evaluate.py

Basic evaluate CLI command / codepath (#2188)

2024-12-16 15:46:31 -05:00

test_cli_fetch.py

CLI cleanup and documentation (#2244)

2025-01-13 17:55:29 +00:00

test_cli_inference.py

CLI cleanup and documentation (#2244)

2025-01-13 17:55:29 +00:00

test_cli_interface.py

CLI cleanup and documentation (#2244)

2025-01-13 17:55:29 +00:00

test_cli_merge_lora.py

CLI cleanup and documentation (#2244)

2025-01-13 17:55:29 +00:00

test_cli_merge_sharded_fsdp_weights.py

CLI cleanup and documentation (#2244)

2025-01-13 17:55:29 +00:00

test_cli_preprocess.py

CLI cleanup and documentation (#2244)

2025-01-13 17:55:29 +00:00

test_cli_sweeps.py

[feature] sweeps (#2171)

2025-02-01 21:11:18 -05:00

test_cli_train.py

add muon optimizer

2025-03-05 10:47:22 -05:00

test_cli_version.py

CLI cleanup and documentation (#2244)

2025-01-13 17:55:29 +00:00

test_utils.py

CLI cleanup and documentation (#2244)

2025-01-13 17:55:29 +00:00