from ultralytics import YOLO

# Load a model
model = YOLO('yolov8s.yaml')

results = model.train(data='data.yaml', save_period=10, epochs=5000, batch=10, imgsz=1504, patience=100, device=1)