diff --git a/test_tipc/test_train_inference_python_xpu.sh b/test_tipc/test_train_inference_python_xpu.sh index 7c6dc1e52a67caf9c858b2f8b6561b3919134b0b..c4add0056c52ff430b2ecb85150b75f13563391e 100644 --- a/test_tipc/test_train_inference_python_xpu.sh +++ b/test_tipc/test_train_inference_python_xpu.sh @@ -29,18 +29,28 @@ fi sed -i 's/use_gpu/use_xpu/g' $FILENAME # disable benchmark as AutoLog required nvidia-smi command sed -i 's/--benchmark:True/--benchmark:False/g' $FILENAME +# python has been updated to version 3.9 for xpu backend +sed -i "s/python3.7/python3.9/g" $FILENAME dataline=`cat $FILENAME` # parser params IFS=$'\n' lines=(${dataline}) +modelname=$(echo ${lines[1]} | cut -d ":" -f2) +if [ $modelname == "rec_r31_sar" ] || [ $modelname == "rec_mtb_nrtr" ]; then + sed -i "s/Global.epoch_num:lite_train_lite_infer=2/Global.epoch_num:lite_train_lite_infer=1/g" $FILENAME + sed -i "s/gpu_list:0|0,1/gpu_list:0,1/g" $FILENAME + sed -i "s/Global.use_xpu:True|True/Global.use_xpu:True/g" $FILENAME +fi + # replace training config file grep -n 'tools/.*yml' $FILENAME | cut -d ":" -f 1 \ | while read line_num ; do train_cmd=$(func_parser_value "${lines[line_num-1]}") trainer_config=$(func_parser_config ${train_cmd}) sed -i 's/use_gpu/use_xpu/g' "$REPO_ROOT_PATH/$trainer_config" + sed -i 's/use_sync_bn: True/use_sync_bn: False/g' "$REPO_ROOT_PATH/$trainer_config" done # change gpu to xpu in execution script