【fix benchmark】 (#681)

* add bencnmark model epochs

* fix benchmark,test=document_fix
Co-authored-by: Nzhengya01 <zhengya01@baidu.com>

test_tipc/configs/FOMM/train_infer_python.txt

@@ -13,7 +13,7 @@ train_infer_img_dir:./data/firstorder_vox_256/test
 null:null
 ##
 trainer:norm_train
-norm_train:tools/main.py -c configs/firstorder_vox_256.yaml --seed 123 -o dataset.train.num_workers=0 log_config.interval=1 snapshot_config.interval=1 dataset.train.num_repeats=1 dataset.train.id_sampling=False
+norm_train:tools/main.py -c configs/firstorder_vox_256.yaml --seed 123 -o log_config.interval=1 snapshot_config.interval=1 dataset.train.num_repeats=1 dataset.train.id_sampling=False
 pact_train:null
 fpgm_train:null
 distill_train:null
@@ -54,6 +54,6 @@ batch_size:16
 fp_items:fp32
 epoch:15
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
-flags:null
+flags:FLAGS_cudnn_exhaustive_search=1
 ===========================infer_benchmark_params==========================
 random_infer_input:[{float32,[3,256,256]}]

test_tipc/configs/StyleGANv2/train_infer_python.txt

@@ -54,6 +54,6 @@ batch_size:8
 fp_items:fp32
 epoch:100
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
-flags:null
+flags:FLAGS_cudnn_exhaustive_search=1
 ===========================infer_benchmark_params==========================
 random_infer_input:[{float32,[1, 512]}, {float32,[1]}]

test_tipc/configs/basicvsr/train_infer_python.txt

@@ -54,6 +54,6 @@ batch_size:2|4
 fp_items:fp32
 total_iters:50
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
-flags:null
+flags:FLAGS_cudnn_exhaustive_search=1
 ===========================infer_benchmark_params==========================
 random_infer_input:[{float32,[6,3,180,320]}]

test_tipc/configs/edvr/train_infer_python.txt

@@ -29,4 +29,4 @@ batch_size:4|64
 fp_items:fp32
 total_iters:100
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
-flags:null
+flags:FLAGS_cudnn_exhaustive_search=1

test_tipc/configs/esrgan/train_infer_python.txt

@@ -13,7 +13,7 @@ train_infer_img_dir:null
 null:null
 ##
 trainer:norm_train
-norm_train:tools/main.py -c configs/esrgan_psnr_x4_div2k.yaml --seed 123 -o log_config.interval=5
+norm_train:tools/main.py -c configs/esrgan_psnr_x4_div2k.yaml --seed 123 -o log_config.interval=5 dataset.train.num_workers=0
 pact_train:null
 fpgm_train:null
 distill_train:null
@@ -29,4 +29,4 @@ batch_size:32|64
 fp_items:fp32
 total_iters:100
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
-flags:null
+flags:FLAGS_cudnn_exhaustive_search=1