USDA-throughput-control/yolov11-pecan/train.py

from ultralytics import YOLO
import torch

device = torch.device('cuda')
# device = 'cpu'

model = YOLO('yolo11s.pt')
model.train(data='./yolov11-pecan/data.yaml',device=device,batch=20,epochs=25,workers=16, imgsz = 1280)