From 5d42cc1b9a90e26b0b9bffba61fae93f5d1691b9 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 2 May 2020 18:44:31 -0700 Subject: [PATCH] Update README.md --- README.md | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 0329a135..06a3ecdd 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