Skip to content

Commit

Permalink
fix benchmark (#596)
Browse files Browse the repository at this point in the history
* fix benchmark

* fix benchmark
  • Loading branch information
mmglove authored Mar 14, 2022
1 parent 7faca16 commit 128a933
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions benchmark/run_benchmark.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down

0 comments on commit 128a933

Please sign in to comment.