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 8d5aff09bd9a5009f51018c592c9405f3e4f1d0e..f290c63b7775743f56df7ad8bc71e43aabcfc2c2 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 49c66fbd0a593fdaafc579d68b639f1aa8d82729..8b689e04db0ffbc2eb935bb95cd80870d21c86ac 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 1796da4122c5aee5e1152bd9794fecacadf80372..2088b195656ba2c52a1379ddc63a1c315d3fc21c 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 783a3372b4a419dd63ae8886723cefd9f28976df..5355648b6dae6d3d5f62b8bc27a1d8bcb55947db 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 d4e8f4497d918c9f58592793067d906c6a3642ab..cf2996a8f93587c127d26a83809edc3d2aa073c2 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 e9c6d6036c149d76cfed812ba6a97db088a332c3..a6bacbd3fa0e5b0c79a4887af0959c3db30a35a9 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