fix googlenet scripts

5 years ago · 6ac99d4a0c
parent 8943d5de34
commit 6ac99d4a0c
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