We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4762551 commit f2114f3Copy full SHA for f2114f3
docs/Train_coco_data.md
@@ -2,6 +2,22 @@
2
3
This guidence shows the training commands for reproducing our results on COCO Dataset.
4
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
21
## For P5 models
22
23
#### YOLOv6-N
0 commit comments