diff --git a/test_tipc/configs/GFPGAN/train_infer_python.txt b/test_tipc/configs/GFPGAN/train_infer_python.txt index 30e4abef4ccfbb8ad24a5dca376ade00181da8f0..e38a7eac883ed608afc613730a03090f4ee7fe1f 100644 --- a/test_tipc/configs/GFPGAN/train_infer_python.txt +++ b/test_tipc/configs/GFPGAN/train_infer_python.txt @@ -49,11 +49,3 @@ null:null null:null --benchmark:False null:null -===========================train_benchmark_params========================== -batch_size:8 -fp_items:fp32 -epoch:100 ---profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile -flags:FLAGS_cudnn_exhaustive_search=1 -===========================infer_benchmark_params========================== -random_infer_input:[{float32,[1, 512]}, {float32,[1]}] diff --git a/test_tipc/configs/aotgan/train_infer_python.txt b/test_tipc/configs/aotgan/train_infer_python.txt index 432baaea5618a668d946b4fc6db5b473a478ba9b..a5d82f988862ff58f4a41196609f3dc52b017fd6 100644 --- a/test_tipc/configs/aotgan/train_infer_python.txt +++ b/test_tipc/configs/aotgan/train_infer_python.txt @@ -49,11 +49,3 @@ null:null null:null --benchmark:True null:null -===========================train_benchmark_params========================== -batch_size:1 -fp_items:fp32 -epoch:10 ---profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile -flags:null -===========================infer_benchmark_params========================== -random_infer_input:[{float32,[3,256,256]}] diff --git a/test_tipc/configs/esrgan/train_infer_python.txt b/test_tipc/configs/esrgan/train_infer_python.txt index 275cf0aa3fc3789ecb7c25aad7100bf33174b17d..dfbb98d92725f1bc74058a206d7a7310b391a911 100644 --- a/test_tipc/configs/esrgan/train_infer_python.txt +++ b/test_tipc/configs/esrgan/train_infer_python.txt @@ -51,7 +51,7 @@ null:null null:null ===========================train_benchmark_params========================== batch_size:32|64 -fp_items:fp32|fp16 +fp_items:fp32 total_iters:500 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile flags:FLAGS_cudnn_exhaustive_search=1