Skip to content

Commit 1ddf692

Browse files
authored
Update readme to remove apex dependency (#727)
* the value should be a list to unpacking * filter the None labels * update readme to remove apex dependency
1 parent 56c2c34 commit 1ddf692

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ Results saved to /content/yolov5/inference/output
9898

9999
## Training
100100

101-
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).
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).
102102
```bash
103103
$ python train.py --data coco.yaml --cfg yolov5s.yaml --weights '' --batch-size 64
104104
yolov5m 48

0 commit comments

Comments
 (0)