From c60d5ae5e015cf127bbb5104eadcc1c64e1af526 Mon Sep 17 00:00:00 2001
From: hwjiaorui <jiaorui3@huawei.com>
Date: Mon, 7 Sep 2020 10:44:59 +0800
Subject: [PATCH] fix bug about target device

---
 model_zoo/official/cv/mobilenetv2_quant/scripts/run_train.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/model_zoo/official/cv/mobilenetv2_quant/scripts/run_train.sh b/model_zoo/official/cv/mobilenetv2_quant/scripts/run_train.sh
index 36594b23dd..51abcda0bd 100644
--- 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 ..