diff --git a/test_tipc/configs/FOMM/train_infer_python.txt b/test_tipc/configs/FOMM/train_infer_python.txt
index 5c1b31029d698ff6a095642fe8d4fdba2364bebc..190ffc260f849256b62da22a0371a6a48f7a2ce1 100644
--- a/test_tipc/configs/FOMM/train_infer_python.txt
+++ b/test_tipc/configs/FOMM/train_infer_python.txt
@@ -52,7 +52,7 @@ null:null
 ===========================train_benchmark_params==========================
 batch_size:16
 fp_items:fp32
-epoch:5
+epoch:15
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
 flags:null
 ===========================infer_benchmark_params==========================