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..3de6add85078c13a323e0f62a8d805880f189c6c 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,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,224,224]}] 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..3a913e253ee31e04d5c3d86b3be34aeb0e777949 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,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,384,384]}] 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..d376d4cba38f52522fb7ecc9a93e38bef1b7b710 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,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,224,224]}] 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..781fefe58dbe18b1bbf89026b59b95c995ecf0c5 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,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,384,384]}] 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..8057bb84b766e0f618b93a85f3f69f82fc8fa712 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,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,224,224]}] 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 a6bacbd3fa0e5b0c79a4887af0959c3db30a35a9..11b2f9dd9eb100c43a87f08192606310c7b555de 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 @@ -56,3 +56,5 @@ 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/config/MobileViT/MobileViT_S_train_infer_python.txt b/test_tipc/config/MobileViT/MobileViT_S_train_infer_python.txt index d45bbb9960325cf9745bda3e63fa3655527d1c9d..d69f26412fbd82c89bdef6095aca27d20df33903 100644 --- a/test_tipc/config/MobileViT/MobileViT_S_train_infer_python.txt +++ b/test_tipc/config/MobileViT/MobileViT_S_train_infer_python.txt @@ -56,3 +56,5 @@ 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,256,256]}] diff --git a/test_tipc/config/MobileViT/MobileViT_XS_train_infer_python.txt b/test_tipc/config/MobileViT/MobileViT_XS_train_infer_python.txt index 826bd4f8bf15496483737e85953b7df61a2f853a..b7436adaaac2a94df9692845b92bc5c46887ba62 100644 --- a/test_tipc/config/MobileViT/MobileViT_XS_train_infer_python.txt +++ b/test_tipc/config/MobileViT/MobileViT_XS_train_infer_python.txt @@ -50,3 +50,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,256,256]}] diff --git a/test_tipc/config/MobileViT/MobileViT_XXS_train_infer_python.txt b/test_tipc/config/MobileViT/MobileViT_XXS_train_infer_python.txt index 83902b67314b74aea1fc63bd94d94a35fcce7cfd..55cb442148ece745be4e2de6d95e2bcaf5fb0074 100644 --- a/test_tipc/config/MobileViT/MobileViT_XXS_train_infer_python.txt +++ b/test_tipc/config/MobileViT/MobileViT_XXS_train_infer_python.txt @@ -50,3 +50,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,256,256]}] diff --git a/test_tipc/config/PVTV2/PVT_V2_B0.txt b/test_tipc/config/PVTV2/PVT_V2_B0_train_infer_python.txt similarity index 94% rename from test_tipc/config/PVTV2/PVT_V2_B0.txt rename to test_tipc/config/PVTV2/PVT_V2_B0_train_infer_python.txt index 1e0a3800c06f9c26a2e1f04167e9fb9996ddee82..a4229d854a9706cfa9bd0baeb18c6c59803197ce 100644 --- a/test_tipc/config/PVTV2/PVT_V2_B0.txt +++ b/test_tipc/config/PVTV2/PVT_V2_B0_train_infer_python.txt @@ -50,3 +50,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,224,224]}] \ No newline at end of file diff --git a/test_tipc/config/PVTV2/PVT_V2_B1.txt b/test_tipc/config/PVTV2/PVT_V2_B1_train_infer_python.txt similarity index 94% rename from test_tipc/config/PVTV2/PVT_V2_B1.txt rename to test_tipc/config/PVTV2/PVT_V2_B1_train_infer_python.txt index 489f8c80d363ba8ae6f98e141e3467e5b7eccb2c..5394790a03b68366a83b4ea2fba3e67c988d7f1e 100644 --- a/test_tipc/config/PVTV2/PVT_V2_B1.txt +++ b/test_tipc/config/PVTV2/PVT_V2_B1_train_infer_python.txt @@ -50,3 +50,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,224,224]}] \ No newline at end of file diff --git a/test_tipc/config/PVTV2/PVT_V2_B2_Linear.txt b/test_tipc/config/PVTV2/PVT_V2_B2_Linear_train_infer_python.txt similarity index 94% rename from test_tipc/config/PVTV2/PVT_V2_B2_Linear.txt rename to test_tipc/config/PVTV2/PVT_V2_B2_Linear_train_infer_python.txt index bf825ddfee4b436999773042f0350d68dce91521..b2aa7df69f36f797f26ab985f379634952fa45d3 100644 --- a/test_tipc/config/PVTV2/PVT_V2_B2_Linear.txt +++ b/test_tipc/config/PVTV2/PVT_V2_B2_Linear_train_infer_python.txt @@ -56,3 +56,5 @@ 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/config/PVTV2/PVT_V2_B2.txt b/test_tipc/config/PVTV2/PVT_V2_B2_train_infer_python.txt similarity index 93% rename from test_tipc/config/PVTV2/PVT_V2_B2.txt rename to test_tipc/config/PVTV2/PVT_V2_B2_train_infer_python.txt index 8d274cd20fcd9cbc2cfaa387f5acf1399330bc8c..d2de07223364b1eb78597c7de97e8a10725bbc18 100644 --- a/test_tipc/config/PVTV2/PVT_V2_B2.txt +++ b/test_tipc/config/PVTV2/PVT_V2_B2_train_infer_python.txt @@ -50,3 +50,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,224,224]}] diff --git a/test_tipc/config/PVTV2/PVT_V2_B3.txt b/test_tipc/config/PVTV2/PVT_V2_B3_train_infer_python.txt similarity index 93% rename from test_tipc/config/PVTV2/PVT_V2_B3.txt rename to test_tipc/config/PVTV2/PVT_V2_B3_train_infer_python.txt index f375fcf2189823bf91a89e87d7d67f7d7047cca9..8f71e148fdf61147d6b4c5acdd818e8051222409 100644 --- a/test_tipc/config/PVTV2/PVT_V2_B3.txt +++ b/test_tipc/config/PVTV2/PVT_V2_B3_train_infer_python.txt @@ -50,3 +50,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,224,224]}] diff --git a/test_tipc/config/PVTV2/PVT_V2_B4.txt b/test_tipc/config/PVTV2/PVT_V2_B4_train_infer_python.txt similarity index 93% rename from test_tipc/config/PVTV2/PVT_V2_B4.txt rename to test_tipc/config/PVTV2/PVT_V2_B4_train_infer_python.txt index f0c7084fb2f70fb030ea331e33731b9511fe005b..304deb6872641c7cb5ae51ccc1aa241fc621e660 100644 --- a/test_tipc/config/PVTV2/PVT_V2_B4.txt +++ b/test_tipc/config/PVTV2/PVT_V2_B4_train_infer_python.txt @@ -50,3 +50,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,224,224]}] diff --git a/test_tipc/config/PVTV2/PVT_V2_B5.txt b/test_tipc/config/PVTV2/PVT_V2_B5_train_infer_python.txt similarity index 93% rename from test_tipc/config/PVTV2/PVT_V2_B5.txt rename to test_tipc/config/PVTV2/PVT_V2_B5_train_infer_python.txt index 9383851c87fb4897cd5301ad279fdcdb678a02f1..9527cb2d385ed217e3212fd535629b2407ff8dee 100644 --- a/test_tipc/config/PVTV2/PVT_V2_B5.txt +++ b/test_tipc/config/PVTV2/PVT_V2_B5_train_infer_python.txt @@ -50,3 +50,5 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran -o Global.benchmark:True null:null null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,224,224]}] diff --git a/test_tipc/config/VAN/VAN_tiny.txt b/test_tipc/config/VAN/VAN_tiny_train_infer_python.txt similarity index 85% rename from test_tipc/config/VAN/VAN_tiny.txt rename to test_tipc/config/VAN/VAN_tiny_train_infer_python.txt index 4ad311b58a653c38e9ae2d656024023102c46737..aafcb4c7fd49ce94a38b32865d1ab002393b2988 100644 --- a/test_tipc/config/VAN/VAN_tiny.txt +++ b/test_tipc/config/VAN/VAN_tiny_train_infer_python.txt @@ -50,9 +50,5 @@ 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 +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,248,248]}]