Commit 30578322 authored by dengjb's avatar dengjb
Browse files

Update train_multi.py

parent 6287a30a
......@@ -4,4 +4,4 @@ from ultralytics import YOLO
model = YOLO("yolov8s.yaml",) # build a new model from scratch
#model = YOLO("yolov8s.pt") # load a pretrained model (recommended for training with a new dataset)
# Use the model
model.train(data="coco.yaml", epochs=100,device=[0, 1, 2, 3])
model.train(data="coco128.yaml", epochs=100,device=[0, 1, 2, 3])
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment