From fa610941002c741690f769f9b873129158584baf Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Fri, 1 Apr 2022 14:34:03 +0000 Subject: [PATCH] Add CSWin benchmark --- .../CSWinTransformer_base_224_train_infer_python.txt | 6 ++++++ .../CSWinTransformer_base_384_train_infer_python.txt | 6 ++++++ .../CSWinTransformer_large_224_train_infer_python.txt | 6 ++++++ .../CSWinTransformer_large_384_train_infer_python.txt | 6 ++++++ .../CSWinTransformer_small_224_train_infer_python.txt | 6 ++++++ .../CSWinTransformer_tiny_224_train_infer_python.txt | 6 ++++++ 6 files changed, 36 insertions(+) diff --git a/test_tipc/config/CSWinTransformer/CSWinTransformer_base_224_train_infer_python.txt b/test_tipc/config/CSWinTransformer/CSWinTransformer_base_224_train_infer_python.txt index 8d5aff09..f290c63b 100644 --- a/test_tipc/config/CSWinTransformer/CSWinTransformer_base_224_train_infer_python.txt +++ b/test_tipc/config/CSWinTransformer/CSWinTransformer_base_224_train_infer_python.txt @@ -50,3 +50,9 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================train_benchmark_params========================== +batch_size:32 +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 diff --git a/test_tipc/config/CSWinTransformer/CSWinTransformer_base_384_train_infer_python.txt b/test_tipc/config/CSWinTransformer/CSWinTransformer_base_384_train_infer_python.txt index 49c66fbd..8b689e04 100644 --- a/test_tipc/config/CSWinTransformer/CSWinTransformer_base_384_train_infer_python.txt +++ b/test_tipc/config/CSWinTransformer/CSWinTransformer_base_384_train_infer_python.txt @@ -50,3 +50,9 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================train_benchmark_params========================== +batch_size:32 +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 diff --git a/test_tipc/config/CSWinTransformer/CSWinTransformer_large_224_train_infer_python.txt b/test_tipc/config/CSWinTransformer/CSWinTransformer_large_224_train_infer_python.txt index 1796da41..2088b195 100644 --- a/test_tipc/config/CSWinTransformer/CSWinTransformer_large_224_train_infer_python.txt +++ b/test_tipc/config/CSWinTransformer/CSWinTransformer_large_224_train_infer_python.txt @@ -50,3 +50,9 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================train_benchmark_params========================== +batch_size:32 +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 diff --git a/test_tipc/config/CSWinTransformer/CSWinTransformer_large_384_train_infer_python.txt b/test_tipc/config/CSWinTransformer/CSWinTransformer_large_384_train_infer_python.txt index 783a3372..5355648b 100644 --- a/test_tipc/config/CSWinTransformer/CSWinTransformer_large_384_train_infer_python.txt +++ b/test_tipc/config/CSWinTransformer/CSWinTransformer_large_384_train_infer_python.txt @@ -50,3 +50,9 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================train_benchmark_params========================== +batch_size:8 +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 diff --git a/test_tipc/config/CSWinTransformer/CSWinTransformer_small_224_train_infer_python.txt b/test_tipc/config/CSWinTransformer/CSWinTransformer_small_224_train_infer_python.txt index d4e8f449..cf2996a8 100644 --- a/test_tipc/config/CSWinTransformer/CSWinTransformer_small_224_train_infer_python.txt +++ b/test_tipc/config/CSWinTransformer/CSWinTransformer_small_224_train_infer_python.txt @@ -50,3 +50,9 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================train_benchmark_params========================== +batch_size:64 +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 diff --git a/test_tipc/config/CSWinTransformer/CSWinTransformer_tiny_224_train_infer_python.txt b/test_tipc/config/CSWinTransformer/CSWinTransformer_tiny_224_train_infer_python.txt index e9c6d603..a6bacbd3 100644 --- a/test_tipc/config/CSWinTransformer/CSWinTransformer_tiny_224_train_infer_python.txt +++ b/test_tipc/config/CSWinTransformer/CSWinTransformer_tiny_224_train_infer_python.txt @@ -50,3 +50,9 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True 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 -- GitLab