diff --git a/test_tipc/config/MobileNetV3_large_x1_0/train_infer_python.txt b/test_tipc/config/MobileNetV3_large_x1_0/train_infer_python.txt
index b784c84dae3790caaea58994f6d366a8c0dce20d..a723ab842c73b8395aed04037c66163bf9cd5dbd 100644
--- a/test_tipc/config/MobileNetV3_large_x1_0/train_infer_python.txt
+++ b/test_tipc/config/MobileNetV3_large_x1_0/train_infer_python.txt
@@ -4,7 +4,7 @@ python:python3.7
 gpu_list:0|0,1
 -o Global.device:gpu
 -o Global.auto_cast:null
--o Global.epochs:lite_train_infer=2|whole_train_infer=120
+-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 Global.pretrained_model:null
@@ -49,4 +49,4 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.save_log_path:null
 -o Global.benchmark:True
 null:null
-null:null
\ No newline at end of file
+null:null