Skip to content

Commit f2114f3

Browse files
author
gengyifei
committed
update yolov6lite train command in README.md
1 parent 4762551 commit f2114f3

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

docs/Train_coco_data.md

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,22 @@
22

33
This guidence shows the training commands for reproducing our results on COCO Dataset.
44

5+
6+
## For Mobile models
7+
8+
#### YOLOv6Lite-S/M/L
9+
10+
```shell
11+
python -m torch.distributed.launch --nproc_per_node 4 tools/train.py \
12+
--batch 128 \
13+
--img_size 416 \ # train with 416 and eval with 320
14+
--conf configs/yolov6_lite/yolov6_lite_s.py \ # yolov6lite_m/l
15+
--data data/coco.yaml \
16+
--epoch 400 \
17+
--device 0,1,2,3 \
18+
--name yolov6_lite_s_coco
19+
```
20+
521
## For P5 models
622

723
#### YOLOv6-N

0 commit comments

Comments
 (0)