Shaun ZhouandClaude Opus 5
865cdd7210
fix(simclr): 续训时改变总轮数不再让学习率卡在 0
resume 会恢复调度器状态，其中包含存档时的 T_max。原计划 10 轮的 cosine 跑完后
学习率已降到 0，此时把 epochs 改成 30 接着练，学习率仍是 0——训练照常进行、
日志照常打印，但一步都学不动。是静默失败，比报错更难发现。
- 总轮数与存档一致时行为不变，直接恢复调度器状态
- 不一致时按新的总轮数重建调度器，并快进到当前 epoch；重建前先把 lr 复位回
initial_lr，因为 cosine 的链式实现基于当前 lr 递推，从 0 出发只会一路留在 0
- 实测续训到 30 轮时 lr 恢复为 7.50e-04，与 cosine 在 10/30 处的理论值一致
- 补三个回归测试：改变总轮数后的学习率、总轮数不变时直接恢复状态、
save_interval 落的中间盘可直接续训
- 修正一个旧测试：它断言续训后 lr 仍等于旧计划末尾的 0，编码的是错误行为
Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
2026-09-04 17:25:45 +08:00
..
2026-09-02 17:55:20 +08:00
2026-09-02 17:55:20 +08:00
2026-09-04 15:09:35 +08:00
2026-09-02 17:55:20 +08:00
2026-09-02 17:55:20 +08:00
2026-09-02 17:55:20 +08:00
2026-09-02 17:55:20 +08:00
2026-09-04 17:25:45 +08:00
2026-09-02 21:43:35 +08:00
2026-09-02 17:55:20 +08:00