feat: 训练与评估入口及 yaml 配置

- train.py: 先过 require_cuda() 守卫再建模型/数据，支持 --resume 与 --set 覆盖 - eval.py: 允许 CPU，输出定位与分类指标并可写 json - configs: model/dataset/train/eval 四类，通过 `_base_` 组合； classes 既可直接写列表，也可指向 YOLO data.yaml 由框架读 names Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
2026-09-02 17:55:20 +08:00
parent ef64e5bf77
commit b0d0dfffbb
9 changed files with 344 additions and 0 deletions
@@ -0,0 +1,8 @@
+_base_:
+  - ./baseline.yaml
+  - ../model/resnet18.yaml
+
+work_dir: runs/resnet18
+train:
+  optimizer:
+    lr: 0.0005