Shaun ZhouandClaude Opus 5 0f5bb4a5c0 feat(cluster): 加 --train-only，自监督训练与聚类可分两步跑 ...

engine/simclr.py 与 engine/cluster.py 本来就互不引用，唯一接口是
simclr_backbone.pth；耦合只在 CLI——method=simclr 会训练完直接聚类。
此前"只聚类"可以用 simclr.checkpoint 做到，"只训练"没有办法。

- --train-only：训练存完权重即退出，不抽特征也不聚类，并打印下一步的聚类命令
- 与 method=pretrained / simclr.checkpoint / --sweep-k 组合时立刻报错说明原因，
  而不是跑完才发现没有产出
- 典型用法：GPU 机器上 --train-only 跑几十小时，之后拿权重反复调 k，每次约 1 分钟

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

2026-09-07 11:38:59 +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_export_clusters.py

feat(scripts): 按聚类结果导出数据集

2026-09-06 13:17:53 +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

feat(cluster): 加 --train-only，自监督训练与聚类可分两步跑

2026-09-07 11:38:59 +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