From e5f0c8b2d21fc7184bc87030b6a312b952df23b8 Mon Sep 17 00:00:00 2001 From: Zhao Yuting <91456992+THUzyt21@users.noreply.github.com> Date: Thu, 15 Sep 2022 14:25:23 +0800 Subject: [PATCH] Update train_infer_python.txt --- tests/test_tipc/configs/mdtc/train_infer_python.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_tipc/configs/mdtc/train_infer_python.txt b/tests/test_tipc/configs/mdtc/train_infer_python.txt index e6a4a2461..883b9c462 100644 --- 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