未验证 提交 128a9331 编写于 作者: G gmm 提交者: GitHub

fix benchmark (#596)

* fix benchmark

* fix benchmark
上级 7faca16a
......@@ -51,6 +51,7 @@ function _train(){
case ${run_mode} in
sp) train_cmd="python -u tools/main.py "${train_cmd} ;;
mp)
rm -rf ./mylog
train_cmd="python -m paddle.distributed.launch --log_dir=./mylog --gpus=$CUDA_VISIBLE_DEVICES tools/main.py "${train_cmd}
log_parse_file="mylog/workerlog.0" ;;
*) echo "choose run_mode(sp or mp)"; exit 1;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册