Browse Source

!3762 fix googlenet script

Merge pull request !3762 from panfengfeng/fix_googlenet_scripts
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
3119aaccfd
1 changed files with 8 additions and 2 deletions
  1. +8
    -2
      model_zoo/official/cv/googlenet/scripts/run_train_gpu.sh

+ 8
- 2
model_zoo/official/cv/googlenet/scripts/run_train_gpu.sh View File

@@ -41,5 +41,11 @@ mkdir ../train
cd ../train || exit

export CUDA_VISIBLE_DEVICES="$2"
mpirun -n $1 --allow-run-as-root \
python3 ${BASEPATH}/../train.py > train.log 2>&1 &

if [ $1 -gt 1 ]
then
mpirun -n $1 --allow-run-as-root \
python3 ${BASEPATH}/../train.py > train.log 2>&1 &
else
python3 ${BASEPATH}/../train.py > train.log 2>&1 &
fi

Loading…
Cancel
Save