import yaml
import os

from ultralytics import YOLO


config = {
    'name': '01_v8x-p26_coco_pretrained',
    'epochs': 500,
    'model': 'yolov8x-p26-c2f.yaml',
    'pretrained': None,
    'patience': 50,
    'seed': 0,
    'deterministic': True,
    'data': 'coco.yaml',
    'imgsz': 640,
    'batch': 64,  # 16
    'optimizer': 'SGD',
    'close_mosaic': 10,
    'mixup': 0.15,
    'mosaic': 1.0,
    'copy_paste': 0.3,
    'scale': 0.9,
    'cos_lr': False,
    'lr0': 0.01,
    'lrf': 0.01,
    'warmup_epochs': 3.0,
    'cache': 'disk',
    'device': [0, 1, 2, 3],
    'workers': 16,
    'project': 'runs/train',
    'val': True,
}

model = YOLO(config['model'])
model.train(**config)