mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix bug about target device

This commit is contained in:
hwjiaorui 2020-09-07 10:44:59 +08:00
parent aaa0436882
commit c60d5ae5e0
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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