From 3080634e50d2bf697fff15771df2397a2477dba0 Mon Sep 17 00:00:00 2001 From: dengkaipeng Date: Wed, 8 Apr 2020 07:10:47 +0000 Subject: [PATCH] fix README --- tsm/README.md | 4 ++-- yolov3/README.md | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tsm/README.md b/tsm/README.md index 8e61389..6bc5c79 100644 --- a/tsm/README.md +++ b/tsm/README.md @@ -73,7 +73,7 @@ python main.py --data= --batch_size=16 使用如下方式进行多卡训练: ```bash -CUDA_VISIBLE_DEVICES=0,1 python main.py --data= --batch_size=8 +CUDA_VISIBLE_DEVICES=0,1 python -m paddle.distributed.launch main.py --data= --batch_size=8 ``` #### 动态图训练 @@ -90,7 +90,7 @@ python main.py --data= --batch_size=16 -d 使用如下方式进行多卡训练: ```bash -CUDA_VISIBLE_DEVICES=0,1 python main.py --data= --batch_size=8 -d +CUDA_VISIBLE_DEVICES=0,1 python -m paddle.distributed.launch main.py --data= --batch_size=8 -d ``` **注意：** 对于静态图和动态图，多卡训练中`--batch_size`为每卡上的batch_size，即总batch_size为`--batch_size`乘以卡数 diff --git a/yolov3/README.md b/yolov3/README.md index 88e1535..cc6d302 100644 --- a/yolov3/README.md +++ b/yolov3/README.md @@ -117,7 +117,7 @@ python main.py --help 使用如下方式进行多卡训练: ```bash -CUDA_VISIBLE_DEVICES=0,1,2,3 python main.py --data= --batch_size=16 +CUDA_VISIBLE_DEVICES=0,1,2,3 python -m paddle.distributed.launch main.py --data= --batch_size=16 ``` #### 动态图训练 @@ -127,7 +127,7 @@ CUDA_VISIBLE_DEVICES=0,1,2,3 python main.py --data= --batch_siz 使用如下方式进行多卡训练: ```bash -CUDA_VISIBLE_DEVICES=0,1,2,3 python main.py --data= --batch_size=16 -d +CUDA_VISIBLE_DEVICES=0,1,2,3 python main.py -m paddle.distributed.launch --data= --batch_size=16 -d ``` -- GitLab