from ultralytics import YOLO

if __name__ == '__main__':
    model = YOLO(r'3cls.yaml')
    results = model.train(
        data='/home/dy/dataset/muju',
        epochs=300,
        imgsz=640,
        batch=4,
        workers=5,
        device='0',
        project='runs/train/muju_cls',
        name='exp_cls',
        exist_ok=False,
        optimizer='AdamW',
        lr0=0.0005,
        patience=0,
    )