File size: 151 Bytes
2cd5935 |
1 2 3 4 5 6 |
from ultralytics import YOLO
model = YOLO("yolov8n.pt")
model.to('cuda')
model.train(data="data.yaml", epochs=100, batch=8, device="cuda", imgsz=640) |
2cd5935 |
1 2 3 4 5 6 |
from ultralytics import YOLO
model = YOLO("yolov8n.pt")
model.to('cuda')
model.train(data="data.yaml", epochs=100, batch=8, device="cuda", imgsz=640) |