From e76f95f14c69a4b01714fe5455ae5dc5662419bb Mon Sep 17 00:00:00 2001
From: huangjun12 <12272008@bjtu.edu.cn>
Date: Thu, 24 Sep 2020 02:42:37 +0000
Subject: [PATCH] refine README when training using single card

---
 dygraph/bmn/README.md | 2 +-
 dygraph/bmn/train.py  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/dygraph/bmn/README.md b/dygraph/bmn/README.md
index c2de8e8c..363287fc 100644
--- a/dygraph/bmn/README.md
+++ b/dygraph/bmn/README.md
@@ -55,7 +55,7 @@ BMN的训练数据采用ActivityNet1.3提供的数据集，我们提供了处理
 
     bash run.sh
 
-若使用单卡训练，请将配置文件bmn.yaml中`TRAIN`和`VALID`对应的num\_gpus调整为1，启动方式如下:
+若使用单卡训练，请将配置文件bmn.yaml中`TRAIN`和`VALID`对应的num\_gpus调整为1，并将train.py文件最后一行的`nprocs`参数值设为1，启动方式如下:
 
     export CUDA_VISIBLE_DEVICES=0
     python train.py
diff --git a/dygraph/bmn/train.py b/dygraph/bmn/train.py
index 21591bb9..bbe5783b 100644
--- a/dygraph/bmn/train.py
+++ b/dygraph/bmn/train.py
@@ -262,4 +262,4 @@ def train_bmn(args):
 
 if __name__ == "__main__":
     args = parse_args()
-    dist.spawn(train_bmn, args=(args, ), nprocs=4)
+    dist.spawn(train_bmn, args=(args, ), nprocs=4)  #nprocs=1 when single card
-- 
GitLab