Skip to content

Commit

Permalink
Update readme to remove apex dependency (#727)
Browse files Browse the repository at this point in the history
* the value should be a list to unpacking

* filter the None labels

* update readme to remove apex dependency
  • Loading branch information
tkianai authored Aug 13, 2020
1 parent 56c2c34 commit 1ddf692
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -98,7 +98,7 @@ Results saved to /content/yolov5/inference/output

## Training

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

1 comment on commit 1ddf692

@glenn-jocher
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

NOTE: v3.0 tag is actually here: 916d4aa

See #907 for comments.

Please sign in to comment.