fcakyon
/

yolov5s-v7.0

Object Detection

computer-vision

Model card Files Files and versions Community

fcakyon commited on Dec 17, 2022

Commit

167924d

·

1 Parent(s): 33d7641

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -62,5 +62,5 @@ results.save(save_dir='results/')
 - Finetune the model on your custom dataset:
 ```bash
-yolov5 train --img 128 --batch 16 --weights fcakyon/yolov5s-v7.0 --epochs 1 --device cpu --freeze 10
 ```

 - Finetune the model on your custom dataset:
 ```bash
+yolov5 train --img 640 --batch 16 --weights fcakyon/yolov5s-v7.0 --epochs 10 --device cuda:0
 ```