mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix npu tipc scripts to reduce running time

pull/2884/head
duanyanhui 2023-07-04 15:45:54 +08:00 committed by cuicheng01
parent a29d48960d
commit a4db6f11bf
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test_tipc/test_train_inference_python.sh
View File

@ -91,7 +91,7 @@ infer_value1=$(func_parser_value "${lines[50]}")
if [ ! $epoch_num ]; then
epoch_num=2
fi
if [[ $MODE = 'benchmark_train' ]]; then
if [[ $MODE = 'benchmark_train' ]] || [[ $train_use_gpu_value = 'npu' ]]; then
epoch_num=1
fi

9
test_tipc/test_train_inference_python_npu.sh
View File

@ -21,6 +21,15 @@ FILENAME=$1
# change gpu to npu in tipc txt configs
sed -i "s/Global.device:gpu/Global.device:npu/g" $FILENAME
sed -i "s/Global.use_gpu/Global.use_npu/g" $FILENAME
sed -i "s/Global.use_tensorrt:True|False/Global.use_tensorrt:False/g" $FILENAME
sed -i "s/Global.save_interval=2/Global.save_interval=1/g" $FILENAME
sed -i "s/-o Global.epochs:lite_train_lite_infer=2/-o Global.epochs:lite_train_lite_infer=1/g" $FILENAME
modelname=$(echo $FILENAME | cut -d '/' -f4)
if [ $modelname == "PVTV2" ] || [ $modelname == "Twins" ] || [ $modelname == "SwinTransformer" ]; then
sed -i "s/gpu_list:0|0,1/gpu_list:0,1/g" $FILENAME
fi
dataline=`cat $FILENAME`
# parser params