diff --git a/README.md b/README.md index 7e39727..35fbee6 100755 --- a/README.md +++ b/README.md @@ -74,9 +74,12 @@ Results saved to /content/yolov5/inference/output ## Reproduce Our Training -Run command below. Training times for yolov5s/m/l/x are 2/4/6/8 days on a single V100 (multi-GPU times faster). +Download [COCO](https://github.com/ultralytics/yolov5/blob/master/data/get_coco2017.sh), install [Apex](https://github.com/NVIDIA/apex) and run command below. Training times for yolov5s/m/l/x are 2/4/6/8 days on a single V100 (multi-GPU times faster). Use the largest `--batch-size` your GPU allows (batch sizes shown for 16 GB devices). ```bash -$ python train.py --data coco.yaml --cfg yolov5s.yaml --weights '' --batch-size 16 +$ python train.py --data coco.yaml --cfg yolov5s.yaml --weights '' --batch-size 64 + yolov5m 48 + yolov5l 32 + yolov5x 16 ```