提交 e76f95f1 编写于 作者: H huangjun12

refine README when training using single card

上级 d8b9c189
......@@ -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
......
......@@ -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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册