v1.0.3

c3b89894 · chenzk · 0086bdff · c3b89894
Commit c3b89894 authored Jun 15, 2024 by chenzk
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

train.py train.py +2 -1

No files found.
--- a/train.py
+++ b/train.py
@@ -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, 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
 # model.push_to_hub("reponame", config={"model": "yolov10n/s/m/b/l/x.yaml"})