From 128a93317d6c61c945986abbba122acdc1f108a1 Mon Sep 17 00:00:00 2001 From: gmm <38800877+mmglove@users.noreply.github.com> Date: Mon, 14 Mar 2022 20:41:16 +0800 Subject: [PATCH] fix benchmark (#596) * fix benchmark * fix benchmark --- benchmark/run_benchmark.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/benchmark/run_benchmark.sh b/benchmark/run_benchmark.sh index a293785..eab29a4 100755 --- a/benchmark/run_benchmark.sh +++ b/benchmark/run_benchmark.sh @@ -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; -- GitLab