AstrAI/scripts/tools
ViperEkura c50adbaac0 feat : replace AdamW with MuonMix (Muon + AdamW) optimizer
- Muon for 2D matrix params, AdamW for 1D (norm/bias/embed)
- MuonMix wrapper handles combined step/zero_grad/state_dict
- New CLI args: weight_decay, muon_momentum, muon_nesterov, muon_ns_steps, muon_adjust_lr
- Removed adamw_beta1/adamw_beta2/adamw_weight_decay
- Moved optimizer/strategy params from signature to **kwargs
2026-07-07 14:10:36 +08:00
..
benchmark.py fix : update benchmark for v1.3.8 cache API, add argparse and cache type switch 2026-07-05 20:30:26 +08:00
generate.py fix: 规范 generate.py 命令行接口 2026-06-19 14:03:02 +08:00
preprocess.py refactor : 清理工厂和配置系统中的死代码与冗余抽象 2026-06-07 11:39:50 +08:00
server.py fix : 修正类型标注与统一 CLI 参数命名 2026-05-27 20:49:44 +08:00
train.py feat : replace AdamW with MuonMix (Muon + AdamW) optimizer 2026-07-07 14:10:36 +08:00