diff --git a/test_tipc/configs/knet/train_infer_python.txt b/test_tipc/configs/knet/train_infer_python.txt
index c2ac2fdcd0..e84d7ea18d 100644
--- a/test_tipc/configs/knet/train_infer_python.txt
+++ b/test_tipc/configs/knet/train_infer_python.txt
@@ -21,13 +21,13 @@ null:null
 null:null
 ##
 ===========================eval_params===========================
-eval:tools/val.py --config test_tipc/configs/knet/knet_resnet50_cityscapes_1024x512_80k --device gpu --num_workers 8
+eval:tools/val.py --config test_tipc/configs/knet/knet_resnet50_cityscapes_1024x512_80k.yml --device gpu --num_workers 8
 null:null
 ##
 ===========================export_params===========================
 --save_dir:
 --model_path:
-norm_export:tools/export.py --config test_tipc/configs/knet/knet_resnet50_cityscapes_1024x512_80k
+norm_export:tools/export.py --config test_tipc/configs/knet/knet_resnet50_cityscapes_1024x512_80k.yml
 quant_export:null
 fpgm_export:null
 distill_export:null
@@ -35,7 +35,7 @@ export1:null
 export2:null
 ===========================infer_params===========================
 infer_model:./test_tipc/output/knet/model.pdparams
-infer_export:tools/export.py --config test_tipc/configs/knet/knet_resnet50_cityscapes_1024x512_80k
+infer_export:tools/export.py --config test_tipc/configs/knet/knet_resnet50_cityscapes_1024x512_80k.yml
 infer_quant:False
 inference:deploy/python/infer.py
 --device:cpu|gpu
diff --git a/test_tipc/configs/segnext/train_infer_python.txt b/test_tipc/configs/segnext/train_infer_python.txt
index fa5d50b313..9e3df3b96f 100644
--- a/test_tipc/configs/segnext/train_infer_python.txt
+++ b/test_tipc/configs/segnext/train_infer_python.txt
@@ -27,7 +27,7 @@ null:null
 ===========================export_params===========================
 --save_dir:
 --model_path:
-norm_export:tools/export.py --config test_tipc/configs/segnext/segnext_mscan_tiny_1024x1024_cityscapes_160k.yml --input_shape 1 3 1024 2048
+norm_export:tools/export.py --config test_tipc/configs/segnext/segnext_mscan_t_cityscapes_1024x1024_160k.yml --input_shape 1 3 1024 2048
 quant_export:null
 fpgm_export:null
 distill_export:null
@@ -35,7 +35,7 @@ export1:null
 export2:null
 ===========================infer_params===========================
 infer_model:./test_tipc/output/segnext/model.pdparams
-infer_export:tools/export.py --config test_tipc/configs/segnext/segnext_mscan_tiny_1024x1024_cityscapes_160k.yml --input_shape 1 3 1024 2048
+infer_export:tools/export.py --config test_tipc/configs/segnext/segnext_mscan_t_cityscapes_1024x1024_160k.yml --input_shape 1 3 1024 2048
 infer_quant:False
 inference:deploy/python/infer.py
 --device:cpu|gpu
diff --git a/test_tipc/test_train_inference_python_npu.sh b/test_tipc/test_train_inference_python_npu.sh
index 009f572bd6..7daffe88a8 100644
--- a/test_tipc/test_train_inference_python_npu.sh
+++ b/test_tipc/test_train_inference_python_npu.sh
@@ -13,25 +13,23 @@ fi
 # change gpu to npu in tipc txt configs
 sed -i "s/Global.use_gpu/Global.use_npu/g" $FILENAME
 sed -i "s/--device gpu/--device npu/g" $FILENAME
+sed -i "s/--device:gpu/--device:npu/g" $FILENAME
 sed -i "s/--device:cpu|gpu/--device:cpu|npu/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 npu backend
+sed -i "s/python3.7/python3.9/g" $FILENAME
 dataline=`cat $FILENAME`
 
 # parser params
 IFS=$'\n'
 lines=(${dataline})
 modelname=$(func_parser_value "${lines[1]}")
-echo $modelname
 
-if  [ $modelname == "hrnet_w48_contrast" ]; then
-    sed -i "s/lite_train_lite_infer=20/lite_train_lite_infer=10/g" $FILENAME
-    sed -i "s/--save_interval 500/--save_interval 250/g" $FILENAME
-fi
-
-if  [ $modelname == "pfpnnet" ]; then
-    sed -i "s/lite_train_lite_infer=20/lite_train_lite_infer=10/g" $FILENAME
-    sed -i "s/--save_interval 500/--save_interval 250/g" $FILENAME
+if  [ $modelname == "hrnet_w48_contrast" ] || [ $modelname == "pfpnnet" ] || [ $modelname == "knet" ] \
+ || [ $modelname == "maskformer" ] || [ $modelname == "ocrnet_hrformer_small" ];then
+    sed -i "s/lite_train_lite_infer=20/lite_train_lite_infer=2/g" $FILENAME
+    sed -i "s/--save_interval 500/--save_interval 50/g" $FILENAME
 fi
 
 # change gpu to npu in execution script