Skip to content

GitLab

Commit c3b89894 authored by chenzk's avatar chenzk
Browse files

v1.0.3

parent 0086bdff
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
train.py
View file @ c3b89894
...@@ -9,7 +9,8 @@ model = YOLOv10() ...@@ -9,7 +9,8 @@ model = YOLOv10()
# model.train(data='coco.yaml', model='yolov10s.yaml', epochs=500, batch=256, imgsz=640, resume=True, device='0,1,2,3') # for resume # model.train(data='coco.yaml', model='yolov10s.yaml', epochs=500, batch=256, imgsz=640, resume=True, device='0,1,2,3') # for resume
model.train(data='coco.yaml', model='yolov10s.yaml', epochs=500, batch=256, imgsz=640, device='0,1,2,3') # model.train(data='coco.yaml', model='yolov10s.yaml', epochs=500, batch=256, imgsz=640, device='0,1,2,3')
model.train(data='coco128.yaml', model='yolov10s.yaml', epochs=500, batch=256, imgsz=640, device='0,1,2,3')
# Note that you can upload your trained model to HuggingFace Hub like below # Note that you can upload your trained model to HuggingFace Hub like below
# model.push_to_hub("reponame", config={"model": "yolov10n/s/m/b/l/x.yaml"}) # model.push_to_hub("reponame", config={"model": "yolov10n/s/m/b/l/x.yaml"})
......
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