Shaun ZhouandClaude Opus 5 004d7a1ce7 fix(simclr): 换优化器种类时不再加载不兼容的动量缓冲 ...

adamw 换成 sgd 时 load_state_dict 不会报错，SGD 会收下 AdamW 的 exp_avg/exp_avg_sq
却永远用不上——动量实际从零开始，但状态里白占内存，之后存盘还会一路带下去。
改为检测到优化器种类变化就跳过加载并给出日志，模型权重与训练进度不受影响。

README 补一张表，说明续训时哪些配置按 yaml、哪些来自存档、哪些不能改。

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>

2026-09-05 11:10:04 +08:00

..

conftest.py

test: 单元测试与端到端冒烟

2026-09-02 17:55:47 +08:00

test_cluster.py

feat(cluster): 输出簇分布图，直观展示簇间距离

2026-09-04 15:09:35 +08:00

test_config.py

feat(cluster): 新增 resnet18 自监督聚类配置，并修复 --set 的 YAML 标量解析

2026-09-04 15:51:28 +08:00

test_dataset.py

test: 单元测试与端到端冒烟

2026-09-02 17:55:47 +08:00

test_device.py

feat: 支持 cuda / mps / cpu 三后端，Mac 上可用 MPS 训练

2026-09-02 21:43:35 +08:00

test_label_mode.py

test: 单元测试与端到端冒烟

2026-09-02 17:55:47 +08:00

test_models.py

fix(backbones): 修复 shufflenet 无法构建，并让换骨架不再踩配置合并的坑

2026-09-04 15:28:49 +08:00

test_profile_inference.py

test: 单元测试与端到端冒烟

2026-09-02 17:55:47 +08:00

test_registry.py

test: 单元测试与端到端冒烟

2026-09-02 17:55:47 +08:00

test_simclr.py

fix(simclr): 换优化器种类时不再加载不兼容的动量缓冲

2026-09-05 11:10:04 +08:00

test_target_encoder.py

test: 单元测试与端到端冒烟

2026-09-02 17:55:47 +08:00

test_train_smoke.py

feat: 支持 cuda / mps / cpu 三后端，Mac 上可用 MPS 训练

2026-09-02 21:43:35 +08:00

test_unlabeled_dataset.py

test: 单元测试与端到端冒烟

2026-09-02 17:55:47 +08:00

test_visualizer.py

test: 单元测试与端到端冒烟

2026-09-02 17:55:47 +08:00

test_yolo_format.py

test: 单元测试与端到端冒烟

2026-09-02 17:55:47 +08:00