diff --git a/test_tipc/configs/DSNet/DSNet_base_train_infer_python.txt b/test_tipc/configs/DSNet/DSNet_base_train_infer_python.txt index 0552622536aed81d57622bffea801aa5ad0d8996..25663f5abc7bd93b931294538dea6b5bf9ec97a7 100644 --- a/test_tipc/configs/DSNet/DSNet_base_train_infer_python.txt +++ b/test_tipc/configs/DSNet/DSNet_base_train_infer_python.txt @@ -50,11 +50,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:False null:null null:null -===========================train_benchmark_params========================== -batch_size:128 -fp_items:fp32 -epoch:1 ---profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile -flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096 ===========================infer_benchmark_params========================== random_infer_input:[{float32,[3,224,224]}] diff --git a/test_tipc/configs/DSNet/DSNet_small_train_infer_python.txt b/test_tipc/configs/DSNet/DSNet_small_train_infer_python.txt index 9454ac48bd5fecb312f66b83c78201a2d7dd2be5..6ef4818eeba8eb8aec5485381c158389a2cd268e 100644 --- a/test_tipc/configs/DSNet/DSNet_small_train_infer_python.txt +++ b/test_tipc/configs/DSNet/DSNet_small_train_infer_python.txt @@ -50,11 +50,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:False null:null null:null -===========================train_benchmark_params========================== -batch_size:128 -fp_items:fp32 -epoch:1 ---profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile -flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096 ===========================infer_benchmark_params========================== random_infer_input:[{float32,[3,224,224]}] diff --git a/test_tipc/configs/DSNet/DSNet_tiny_train_infer_python.txt b/test_tipc/configs/DSNet/DSNet_tiny_train_infer_python.txt index 0ae29de5c5c85d1134a8a24b93102cb42dae9f13..f5dd4b310ee764cec4eea2382f70baca34b9e964 100644 --- a/test_tipc/configs/DSNet/DSNet_tiny_train_infer_python.txt +++ b/test_tipc/configs/DSNet/DSNet_tiny_train_infer_python.txt @@ -50,11 +50,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:False null:null null:null -===========================train_benchmark_params========================== -batch_size:128 -fp_items:fp32 -epoch:1 ---profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile -flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096 ===========================infer_benchmark_params========================== random_infer_input:[{float32,[3,224,224]}]