diff --git a/test_tipc/config/EfficientNet/EfficientNetB6_train_infer_python.txt b/test_tipc/config/EfficientNet/EfficientNetB6_train_infer_python.txt
index 5af97fb15..5b5b7833e 100644
--- a/test_tipc/config/EfficientNet/EfficientNetB6_train_infer_python.txt
+++ b/test_tipc/config/EfficientNet/EfficientNetB6_train_infer_python.txt
@@ -6,7 +6,7 @@ gpu_list:0|0,1
 -o Global.auto_cast:null
 -o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120
 -o Global.output_dir:./output/
--o DataLoader.Train.sampler.batch_size:8
+-o DataLoader.Train.sampler.batch_size:2
 -o Global.pretrained_model:null
 train_model_name:latest
 train_infer_img_dir:./dataset/ILSVRC2012/val
diff --git a/test_tipc/config/EfficientNet/EfficientNetB7_train_infer_python.txt b/test_tipc/config/EfficientNet/EfficientNetB7_train_infer_python.txt
index 669fbe230..1dbdf6eee 100644
--- a/test_tipc/config/EfficientNet/EfficientNetB7_train_infer_python.txt
+++ b/test_tipc/config/EfficientNet/EfficientNetB7_train_infer_python.txt
@@ -6,7 +6,7 @@ gpu_list:0|0,1
 -o Global.auto_cast:null
 -o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120
 -o Global.output_dir:./output/
--o DataLoader.Train.sampler.batch_size:8
+-o DataLoader.Train.sampler.batch_size:2
 -o Global.pretrained_model:null
 train_model_name:latest
 train_infer_img_dir:./dataset/ILSVRC2012/val
diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh
index 0577a0309..23b2f70a1 100644
--- a/test_tipc/test_train_inference_python.sh
+++ b/test_tipc/test_train_inference_python.sh
@@ -23,7 +23,7 @@ epoch_key=$(func_parser_key "${lines[6]}")
 epoch_num=$(func_parser_params "${lines[6]}")
 save_model_key=$(func_parser_key "${lines[7]}")
 train_batch_key=$(func_parser_key "${lines[8]}")
-train_batch_value=$(func_parser_params "${lines[8]}")
+train_batch_value=$(func_parser_value "${lines[8]}")
 pretrain_model_key=$(func_parser_key "${lines[9]}")
 pretrain_model_value=$(func_parser_value "${lines[9]}")
 train_model_name=$(func_parser_value "${lines[10]}")
@@ -274,9 +274,9 @@ else
                 fi
                 
                 # load pretrain from norm training if current trainer is pact or fpgm trainer
-                if [ ${trainer} = ${pact_key} ] || [ ${trainer} = ${fpgm_key} ]; then
-                    set_pretrain="${load_norm_train_model}"
-                fi
+                # if [ ${trainer} = ${pact_key} ] || [ ${trainer} = ${fpgm_key} ]; then
+                #    set_pretrain="${load_norm_train_model}"
+                # fi
 
                 set_save_model=$(func_set_params "${save_model_key}" "${save_log}")
                 if [ ${#gpu} -le 2 ];then  # train with cpu or single gpu