from ultralytics import YOLO

model = YOLO('/home/hx/yolo/ultralytics_yolo11-main/yolo11n.pt')

# 冻结 DFL 层，防止权重过大
for name, param in model.model.named_parameters():
    if "dfl.conv" in name:
        param.requires_grad = False

# 训练
results = model.train(
    data='/home/hx/yolo/ultralytics_yolo11-main/ailaidata.yaml',
    epochs=500,
    imgsz=640,
    batch=16,
    workers=4,
    device='0',
    project='runs/train',
    name='exp_ailai_detect',
    exist_ok=False,
    optimizer='AdamW',
    lr0=0.0001,
    lrf=0.01,
    patience=50,
    amp=True,
    close_mosaic=10,
    val=True,
)