Upload config/train_comparison.yaml with huggingface_hub

config/train_comparison.yaml

@@ -0,0 +1,66 @@
+# 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