diff --git a/test_tipc/configs/DLA/DLA169_train_infer_python.txt b/test_tipc/configs/DLA/DLA169_train_infer_python.txt index e01a74cc3928be9bbc44d14bedc9b30d8231062d..a66462028130541a37ec85220670bf2dd398ded9 100644 --- a/test_tipc/configs/DLA/DLA169_train_infer_python.txt +++ b/test_tipc/configs/DLA/DLA169_train_infer_python.txt @@ -50,5 +50,12 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o Global.benchmark:False null:null null:null +===========================train_benchmark_params========================== +batch_size:64 +fp_items:fp32|fp16 +epoch:1 +model_type:norm_train|to_static_train +--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]}] \ No newline at end of file diff --git a/test_tipc/configs/DeiT/DeiT_base_patch16_224_train_infer_python.txt b/test_tipc/configs/DeiT/DeiT_base_patch16_224_train_infer_python.txt index fb44f6a7380239828bcdfa6bca91aab87af2c0a3..9877a1d44d36518cf4765f3478eab3f74038533d 100644 --- a/test_tipc/configs/DeiT/DeiT_base_patch16_224_train_infer_python.txt +++ b/test_tipc/configs/DeiT/DeiT_base_patch16_224_train_infer_python.txt @@ -50,5 +50,12 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o Global.benchmark:False null:null null:null +===========================train_benchmark_params========================== +batch_size:256 +fp_items:fp32|fp16 +epoch:1 +model_type:norm_train|to_static_train +--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]}] \ No newline at end of file +random_infer_input:[{float32,[3,224,224]}] diff --git a/test_tipc/configs/DeiT/DeiT_tiny_patch16_224_train_infer_python.txt b/test_tipc/configs/DeiT/DeiT_tiny_patch16_224_train_infer_python.txt index b0d4efd1c4d1fdf45216b9092f1c16b117f55a43..cc60e29e2de4eac9a8497061797225933ce4f454 100644 --- a/test_tipc/configs/DeiT/DeiT_tiny_patch16_224_train_infer_python.txt +++ b/test_tipc/configs/DeiT/DeiT_tiny_patch16_224_train_infer_python.txt @@ -50,5 +50,12 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o Global.benchmark:False null:null null:null +===========================train_benchmark_params========================== +batch_size:256 +fp_items:fp32|fp16 +epoch:1 +model_type:norm_train|to_static_train +--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]}] \ No newline at end of file +random_infer_input:[{float32,[3,224,224]}] diff --git a/test_tipc/configs/GhostNet/GhostNet_x1_3_train_infer_python.txt b/test_tipc/configs/GhostNet/GhostNet_x1_3_train_infer_python.txt index 769761b16207bc190b335ac2ebf7cb60b46a1259..6e3f56d6868efdfe83485070876d99376c932ffe 100644 --- a/test_tipc/configs/GhostNet/GhostNet_x1_3_train_infer_python.txt +++ b/test_tipc/configs/GhostNet/GhostNet_x1_3_train_infer_python.txt @@ -50,5 +50,12 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o Global.benchmark:False null:null null:null +===========================train_benchmark_params========================== +batch_size:512 +fp_items:fp32|fp16 +epoch:1 +model_type:norm_train|to_static_train +--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]}] \ No newline at end of file diff --git a/test_tipc/configs/MixNet/MixNet_L_train_infer_python.txt b/test_tipc/configs/MixNet/MixNet_L_train_infer_python.txt index 66e9170f25a7301ecbbcf6404a6a3f155091fdfe..400796ff1558b983ef2323803824f6de19c5e42c 100644 --- a/test_tipc/configs/MixNet/MixNet_L_train_infer_python.txt +++ b/test_tipc/configs/MixNet/MixNet_L_train_infer_python.txt @@ -50,5 +50,12 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o Global.benchmark:False null:null null:null +===========================train_benchmark_params========================== +batch_size:64 +fp_items:fp32|fp16 +epoch:1 +model_type:norm_train|to_static_train +--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]}] \ No newline at end of file diff --git a/test_tipc/configs/ReXNet/ReXNet_1_0_train_infer_python.txt b/test_tipc/configs/ReXNet/ReXNet_1_0_train_infer_python.txt index 63045f7d263cbd3d2bbfeee893795e927b953fb7..57c3429c694ef5ca52607aa7b9e5a348111613ca 100644 --- a/test_tipc/configs/ReXNet/ReXNet_1_0_train_infer_python.txt +++ b/test_tipc/configs/ReXNet/ReXNet_1_0_train_infer_python.txt @@ -50,5 +50,12 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o Global.benchmark:False null:null null:null +===========================train_benchmark_params========================== +batch_size:64 +fp_items:fp32|fp16 +epoch:1 +model_type:norm_train|to_static_train +--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]}] \ No newline at end of file diff --git a/test_tipc/configs/Twins/alt_gvt_small_train_infer_python.txt b/test_tipc/configs/Twins/alt_gvt_small_train_infer_python.txt index d174ca3f0226a36267e64a13b4978710c2c01de4..df3ad865329370992b0218900b32caae7acd931e 100644 --- a/test_tipc/configs/Twins/alt_gvt_small_train_infer_python.txt +++ b/test_tipc/configs/Twins/alt_gvt_small_train_infer_python.txt @@ -54,7 +54,7 @@ null:null batch_size:128 fp_items:fp32|fp16 epoch:1 -model_type:norm_train +model_type:norm_train|to_static_train --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========================== diff --git a/test_tipc/configs/Twins/pcpvt_small_train_infer_python.txt b/test_tipc/configs/Twins/pcpvt_small_train_infer_python.txt index 61e25436ef3f0400748ba311efbc0247d3ccb66d..9af3deb2f12513c428ce34f773f0c9933243d26b 100644 --- a/test_tipc/configs/Twins/pcpvt_small_train_infer_python.txt +++ b/test_tipc/configs/Twins/pcpvt_small_train_infer_python.txt @@ -50,5 +50,12 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o Global.benchmark:False null:null null:null +===========================train_benchmark_params========================== +batch_size:128 +fp_items:fp32|fp16 +epoch:1 +model_type:norm_train|to_static_train +--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]}] \ No newline at end of file