Skip to content

GitLab

Commit 1efe33af authored by dengjb's avatar dengjb
Browse files

update train.py

parent b273d601
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
train.py
View file @ 1efe33af
...@@ -4,4 +4,4 @@ from ultralytics import YOLO ...@@ -4,4 +4,4 @@ from ultralytics import YOLO
model = YOLO("yolov8s.yaml") # build a new model from scratch 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) #model = YOLO("yolov8s.pt") # load a pretrained model (recommended for training with a new dataset)
# Use the model # Use the model
model.train(data="coco.yaml", epochs=100,verbose=True) model.train(data="coco128.yaml", epochs=100,verbose=True)
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