Browse Source

fix bug about target device

tags/v1.0.0
hwjiaorui 5 years ago
parent
commit
c60d5ae5e0
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      model_zoo/official/cv/mobilenetv2_quant/scripts/run_train.sh

+ 2
- 2
model_zoo/official/cv/mobilenetv2_quant/scripts/run_train.sh View File

@@ -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 ..


Loading…
Cancel
Save