training for JC deployment
This commit is contained in:
8
yolov11-pecan/train.py
Normal file
8
yolov11-pecan/train.py
Normal file
@@ -0,0 +1,8 @@
|
||||
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)
|
||||
Reference in New Issue
Block a user