Merge branch 'meituan:main' into main

ZHEQIUSHUI · Jun 26, 2022 · ca2b911 · ca2b911
2 parents de55bf1 + d22979c
commit ca2b911
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -47,7 +47,7 @@ python tools/infer.py --weights yolov6s.pt --source img.jpg / imgdir
 Single GPU
 
 ```shell
-python tools/train.py --batch 256 --conf configs/yolov6s.py --data data/coco.yaml --device 0
+python tools/train.py --batch 32 --conf configs/yolov6s.py --data data/coco.yaml --device 0
                                          configs/yolov6n.py
 ```