_base_:
  - ../model/tiny_conv.yaml
  - ../dataset/film_defect.yaml

work_dir: runs/baseline
seed: 42
deterministic: false
device: auto            # auto 按 cuda > mps > cpu 挑最快的可用后端；也可写 cuda / mps / cpu

train:
  epochs: 100
  amp: auto             # auto: CUDA 上开、MPS 上关（实测 MPS autocast 比 fp32 慢）；也可写 true / false
  clip_grad: 5.0
  log_interval: 20
  monitor: f1           # 用于挑选 best.pth 的验证指标
  restart_schedule: false   # 续训时按剩余轮数重开学习率曲线（调度已到末尾、想调大 lr 时用）

  optimizer:
    name: adamw
    lr: 0.001
    weight_decay: 0.0001

  scheduler:
    name: cosine

  loss:
    heatmap_weight: 1.0
    wh_weight: 0.1
    offset_weight: 1.0
    cls_weight: 1.0

  eval:
    score_threshold: 0.3
    iou_threshold: 0.5
    topk: 50

  # 训练过程可视化（SwanLab）。未安装或 enabled=false 时自动跳过。
  swanlab:
    enabled: true
    project: film-defect
    name: null            # 缺省用 work_dir 的目录名
    mode: online          # online(云端) | local(本地看板) | offline(仅存盘) | disabled
    tags: [tiny_conv, center-head]