from ultralytics import YOLO

if __name__ == "__main__":
    # Load YOLOv8n (nano) model for training
    model = YOLO("yolo11n-pose.pt")  # Starts from pretrained weights

    # Train
    model.train(
        data="data/data.yaml", 
        epochs=50,
        imgsz=640,
        batch=16,
        name="yolo11n_lazyUser",
        device=0,            
        mosaic=0, 
    )