zjsh_yolov11/ultralytics_yolo11-main/train_ailai_main.py

from ultralytics import YOLO

if __name__ == '__main__':
    #model = YOLO(r'/home/hx/yolo/ultralytics_yolo11-main/ultralytics/cfg/models/11/yolo11-obb.yaml')
    #model = YOLO(r'/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai2/weights/best.pt')
    model = YOLO(r'/home/hx/yolo/ultralytics_yolo11-main/ultralytics/cfg/models/11/yolo11-ailai.yaml')
    results = model.train(
        data='data_ailai.yaml',
        epochs=300,
        imgsz=640,
        batch=4,
        workers=10,
        device='0',
        project='runs/train',
        name='exp_ailai',
        exist_ok=False,
        optimizer='AdamW',
        lr0=0.0005,
        patience=0,
    )