Merge pull request #187 from Jackwaterveg/CICE_update

fix bug of test.sh
PaddlePaddle · Sep 29, 2021 · 092e108 · 092e108
2 parents bc36816 + 09a6a0d
commit 092e108
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test.sh b/tests/test.sh
@@ -321,9 +321,9 @@ else
                 if [ ${#gpu} -le 2 ];then  # train with cpu or single gpu
                     cmd="${python} ${run_train} "
                 elif [ ${#gpu} -le 15 ];then  # train with multi-gpu
-                    gsu=${gru//,/ }
+                    gsu=${gpu//,/ }
                     nump=`echo $gsu | wc -w`
-                    cmd="${python} ${run_train} --nproc=$nump"
+                    cmd="${python} ${run_train} --nprocs=$nump"
                 else     # train with multi-machine
                     cmd="${python} -m paddle.distributed.launch --ips=${ips} --gpus=${gpu} ${run_train} ${set_save_model} ${set_pretrain} ${set_epoch} ${set_autocast} ${set_batchsize} ${set_train_params1}"
                 fi