Shaun ZhouandClaude Opus 5 b6f83fcad0 fix(simclr): 周期性落盘，中途中断不再丢失训练成果 ...

此前 fit() 只在全部 epoch 跑完后才保存骨架，一次跑到 epoch 12 被停掉后
13 分钟的训练全部作废。

- 每 save_interval（默认 5）个 epoch 保存一次骨架
- 捕获 KeyboardInterrupt，中断时先落盘再返回
- 聚类脚本改用 resolve_device，自监督分支不再有 GPU 守卫

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

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

..

__init__.py

feat(engine): 训练评估循环、CenterNet 损失与指标

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

checkpoint.py

feat(engine): 训练评估循环、CenterNet 损失与指标

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

cluster.py

feat: 未分类样本的无监督聚类（多 stage 特征 + KMeans，可选 SimCLR）

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

evaluator.py

feat(engine): 训练评估循环、CenterNet 损失与指标

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

losses.py

feat(engine): 训练评估循环、CenterNet 损失与指标

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

metrics.py

feat(engine): 训练评估循环、CenterNet 损失与指标

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

optim.py

feat(engine): 训练评估循环、CenterNet 损失与指标

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

simclr.py

fix(simclr): 周期性落盘，中途中断不再丢失训练成果

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

trainer.py

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

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

visualizer.py

feat(engine): 训练评估循环、CenterNet 损失与指标

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