dgwon/yolov8s-coco2017-vehicle-detection-2

results = model.train(
    data=yaml_path,
    epochs=100,
    imgsz=640,
    batch=96,  # bigger batch size
    device=0,
    project=wandb_project_name,
    name=run_name,
    save=True,
    save_period=10,
    patience=10,
    workers=16,

    optimizer="AdamW",
    lr0=1.0e-3,  # learning rate used in yolo pretraining
    lrf=0.05,
    cos_lr=True,
    warmup_epochs=3,
    weight_decay=5e-4,
    amp=True,

    close_mosaic=10,
    mosaic=1.0,
    mixup=0.0,
    copy_paste=0.05,

    fliplr=0.5,
    flipud=0.0,
    hsv_h=0.02,
    hsv_s=0.7,
    hsv_v=0.4,
)