from ultralytics import YOLO


config = {
    # project
    'name': 'v11x-p2_coco_pretrained',
    'model': 'yolo11x-p2.yaml',
    'pretrained': None,

    # hyperparams
    'lr0': 0.01,
    'lrf': 0.01,
    'momentum': 0.937,
    'weight_decay': 0.0005,
    'epochs': 600,
    'imgsz': 640,
    'batch': 48,
    'cos_lr': False,
    'close_mosaic': 10,
    'warmup_epochs': 3.0,
    'warmup_momentum': 0.8,
    'warmup_bias_lr': 0.0,
    'box': 7.5,
    'cls': 0.5,
    'dfl': 1.5,
    'hsv_h': 0.015,
    'hsv_s': 0.7,
    'hsv_v': 0.4,
    'translate': 0.1,
    'scale': 0.9,  # coco=0.9
    'fliplr': 0.5,
    'optimizer': 'SGD',
    'patience': 100,
    # augment: https://github.com/ultralytics/ultralytics/issues/16607
    'mixup': 0.15,
    'mosaic': 1.0,
    'copy_paste': 0.3,

    # system
    'seed': 0,
    'deterministic': True,
    'data': 'ultralytics/cfg/datasets/coco.yaml',
    'cache': 'disk',
    'device': [0, 1, 2, 3],
    'workers': 16,
    'project': 'runs/train',
    'val': True,
}

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