Update train_infer_python.txt

3 years ago · e5f0c8b2d2
parent c81d95a837
commit e5f0c8b2d2
1 changed files with 1 additions and 1 deletions
--- a/tests/test_tipc/configs/mdtc/train_infer_python.txt
+++ b/tests/test_tipc/configs/mdtc/train_infer_python.txt
@ -52,6 +52,6 @@ null:null
 ===========================train_benchmark_params==========================
 batch_size:16|30
 fp_items:fp32
-iteration:20 #modify max_epoch here
+iteration:10 #modify max_epoch here and 10 epochs runs merely 5 minutes.
 --profiler-options:"batch_range=[10,35];state=GPU;tracer_option=Default;profile_path=model.profile"
 flags:null