From 1ddf692bf89ac47d5e2dc8333cd0f4f145bb7f61 Mon Sep 17 00:00:00 2001 From: tkianai <49858938+tkianai@users.noreply.github.com> Date: Thu, 13 Aug 2020 12:18:29 +0800 Subject: [PATCH] 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 --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 3540d52db0cc..fa67b7dc682e 100755 --- a/README.md +++ b/README.md @@ -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