| from ultralytics import YOLO | |
| model = YOLO('yolo11m.pt') | |
| model.train( | |
| data='/content/cigarette-3/data.yaml', | |
| epochs=30, | |
| imgsz=640, | |
| batch=16, | |
| patience=5, | |
| optimizer="SGD", | |
| lr0=0.001, | |
| lrf=0.01, | |
| momentum=0.937, | |
| weight_decay=0.0005, | |
| warmup_epochs=3, | |
| hsv_h=0.015, hsv_s=0.7, hsv_v=0.4, | |
| translate=0.1, scale=0.5, | |
| fliplr=0.5, mosaic=1.0, mixup=0.1, | |
| device=0, | |
| cache=True | |
| ) |