diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh
index 4e21699745b0e373c1d14a6fbae4c16404ca034b..b5563d71d4e3c04f4c4023535e92a5710419859e 100644
--- a/test_tipc/test_train_inference_python.sh
+++ b/test_tipc/test_train_inference_python.sh
@@ -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
 
diff --git a/test_tipc/test_train_inference_python_npu.sh b/test_tipc/test_train_inference_python_npu.sh
index e933eff5b3518e717de503932a901efdfcfd17c9..60071197844a759fad96f18bf438625484c2996a 100644
--- a/test_tipc/test_train_inference_python_npu.sh
+++ b/test_tipc/test_train_inference_python_npu.sh
@@ -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