diff --git a/test_tipc/config/MobileViT/MobileViT_S_train_infer_python.txt b/test_tipc/config/MobileViT/MobileViT_S_train_infer_python.txt
index 950f3ee4ffe719120145805a9934c5bb93279914..d45bbb9960325cf9745bda3e63fa3655527d1c9d 100644
--- a/test_tipc/config/MobileViT/MobileViT_S_train_infer_python.txt
+++ b/test_tipc/config/MobileViT/MobileViT_S_train_infer_python.txt
@@ -50,3 +50,9 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran
 -o Global.benchmark:True
 null:null
 null:null
+===========================train_benchmark_params==========================
+batch_size:128
+fp_items:fp32
+epoch:1
+--profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
+flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096
diff --git a/test_tipc/config/MobileViT/MobileViT_XS_train_infer_python.txt b/test_tipc/config/MobileViT/MobileViT_XS_train_infer_python.txt
index 826bd4f8bf15496483737e85953b7df61a2f853a..a366d42c44832ab30a1c5f9f1b2b7eea36be63fe 100644
--- a/test_tipc/config/MobileViT/MobileViT_XS_train_infer_python.txt
+++ b/test_tipc/config/MobileViT/MobileViT_XS_train_infer_python.txt
@@ -50,3 +50,9 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran
 -o Global.benchmark:True
 null:null
 null:null
+===========================train_benchmark_params==========================
+batch_size:128
+fp_items:fp32
+epoch:1
+--profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
+flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096
diff --git a/test_tipc/config/MobileViT/MobileViT_XXS_train_infer_python.txt b/test_tipc/config/MobileViT/MobileViT_XXS_train_infer_python.txt
index 83902b67314b74aea1fc63bd94d94a35fcce7cfd..cc7169416d858dc8b525ad29d4ff1fc8683dc981 100644
--- a/test_tipc/config/MobileViT/MobileViT_XXS_train_infer_python.txt
+++ b/test_tipc/config/MobileViT/MobileViT_XXS_train_infer_python.txt
@@ -50,3 +50,9 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran
 -o Global.benchmark:True
 null:null
 null:null
+===========================train_benchmark_params==========================
+batch_size:128
+fp_items:fp32
+epoch:1
+--profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
+flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096