diff --git a/README.md b/README.md index 0329a13579..06a3ecddae 100755 --- a/README.md +++ b/README.md @@ -155,11 +155,12 @@ Speed: 17.5/2.3/19.9 ms inference/NMS/total per 640x640 image at batch-size 16 # Reproduce Our Results -This command trains `yolov3-spp.cfg` from scratch to our mAP above. Training takes about one week on a 2080Ti. +Run commands below. Training takes about one week on a 2080Ti per model. ```bash -$ python3 train.py --weights '' --cfg yolov3-spp.cfg --epochs 300 --batch-size 16 --img 320 640 +$ python train.py --data coco2014.data --weights '' --batch-size 16 --cfg yolov3-spp.cfg +$ python train.py --data coco2014.data --weights '' --batch-size 32 --cfg yolov3-tiny.cfg ``` - + # Reproduce Our Environment