提交 c60d5ae5 编写于 作者: H hwjiaorui

fix bug about target device

上级 aaa04368
......@@ -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 ..
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册