invoice-layout-yolov8m / config /train_comparison.yaml
AvoCahDoe's picture
Upload config/train_comparison.yaml with huggingface_hub
45179c8 verified
Raw
History Blame Contribute Delete
1.1 kB
# 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