# Multi-model YOLO comparison training (optimized for RTX 4070 Laptop)
epochs: 100
patience: 25
device: 0
workers: 0
cache: disk
project: runs/comparison

# Optimizer
optimizer: AdamW
cos_lr: true
lr0: 0.01
lrf: 0.01
momentum: 0.937
weight_decay: 0.0005
warmup_epochs: 3.0

# Loss weights
box: 7.5
cls: 0.5
dfl: 1.5

# Geometric augmentation
mosaic: 1.0
mixup: 0.15
copy_paste: 0.1
scale: 0.5
perspective: 0.0005
degrees: 3.0
translate: 0.1
fliplr: 0.0
flipud: 0.0
close_mosaic: 10

# Visual augmentation
hsv_h: 0.015
hsv_s: 0.7
hsv_v: 0.4
erasing: 0.4

# Per-model overrides — train smallest first (n → s → m → 11m → v8x → 11x)
models:
  yolov8n:
    weights: yolov8n.pt
    batch: 8
    imgsz: 1024
  yolov8s:
    weights: yolov8s.pt
    batch: 8
    imgsz: 1024
  yolov8m:
    weights: yolov8m.pt
    batch: 6
    imgsz: 1024
  yolo11m:
    weights: yolo11m.pt
    batch: 6
    imgsz: 1024
  yolov8x:
    weights: yolov8x.pt
    batch: 2
    imgsz: 1024
  yolo11x:
    weights: yolo11x.pt
    batch: 2
    imgsz: 1024