fix googlenet scripts

6ac99d4a · panfengfeng · 8943d5de · 6ac99d4a
隐藏空白更改
内联并排

Showing with 8 addition and 2 deletion

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

未找到文件。
--- 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