!3762 fix googlenet script

Merge pull request !3762 from panfengfeng/fix_googlenet_scripts
5 years ago · 3119aaccfd
parent 641cb4ac6d 6ac99d4a0c
commit 3119aaccfd
1 changed files with 8 additions and 2 deletions
--- a/model_zoo/official/cv/googlenet/scripts/run_train_gpu.sh
+++ b/model_zoo/official/cv/googlenet/scripts/run_train_gpu.sh
@ -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