tkianai
commited on
Update readme to remove apex dependency (#727)
Browse files* the value should be a list to unpacking
* filter the None labels
* update readme to remove apex dependency
README.md
CHANGED
@@ -98,7 +98,7 @@ Results saved to /content/yolov5/inference/output
|
|
98 |
|
99 |
## Training
|
100 |
|
101 |
-
Download [COCO](https://github.com/ultralytics/yolov5/blob/master/data/get_coco2017.sh)
|
102 |
```bash
|
103 |
$ python train.py --data coco.yaml --cfg yolov5s.yaml --weights '' --batch-size 64
|
104 |
yolov5m 48
|
|
|
98 |
|
99 |
## Training
|
100 |
|
101 |
+
Download [COCO](https://github.com/ultralytics/yolov5/blob/master/data/get_coco2017.sh) 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).
|
102 |
```bash
|
103 |
$ python train.py --data coco.yaml --cfg yolov5s.yaml --weights '' --batch-size 64
|
104 |
yolov5m 48
|