fix bug about target device

5 years ago · c60d5ae5e0
parent aaa0436882
commit c60d5ae5e0
1 changed files with 2 additions and 2 deletions
--- a/model_zoo/official/cv/mobilenetv2_quant/scripts/run_train.sh
+++ b/model_zoo/official/cv/mobilenetv2_quant/scripts/run_train.sh
@ -227,7 +227,7 @@ run_ascend(){
        
        if [ $# == 4 ]
        then
-            python train.py --device_traget=$1  --dataset_path=$PATH2 --pre_trained=$PATH3 &> train.log &
+            python train.py --device_target=$1  --dataset_path=$PATH2 --pre_trained=$PATH3 &> train.log &
        fi

        cd ../.. || exit
@ -284,7 +284,7 @@ run_gpu(){
    if [ $# == 4 ]
    then
        mpirun --allow-run-as-root -n ${RANK_SIZE} \
-        python train.py --device_traget=$1  --dataset_path=$PATH1 --pre_trained=$PATH2 &> train.log &
+        python train.py --device_target=$1  --dataset_path=$PATH1 --pre_trained=$PATH2 &> train.log &
    fi

    cd ..