from ultralytics import YOLO

def main():
    model = YOLO("models/yolo11n.pt")
    model.train(
        data="./datasets/unified/data.yaml",
        epochs=200,
        patience=30,
        imgsz=640,
        batch=12,
        device=0,
        project="result",
        name="unified-training",
        exist_ok=True
    )

if __name__ == "__main__":
    main()