diff --git a/test_tipc/configs/MobileViT/MobileViT_S_train_infer_python.txt b/test_tipc/configs/MobileViT/MobileViT_S_train_infer_python.txt index a8bda5b6d..1aae55af5 100644 --- a/test_tipc/configs/MobileViT/MobileViT_S_train_infer_python.txt +++ b/test_tipc/configs/MobileViT/MobileViT_S_train_infer_python.txt @@ -52,9 +52,9 @@ null:null null:null ===========================train_benchmark_params========================== batch_size:128 -fp_items:fp32 +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/PPLCNet/PPLCNet_x0_75_train_infer_python.txt b/test_tipc/configs/PPLCNet/PPLCNet_x0_75_train_infer_python.txt index 8b3be58ca..048ace641 100644 --- a/test_tipc/configs/PPLCNet/PPLCNet_x0_75_train_infer_python.txt +++ b/test_tipc/configs/PPLCNet/PPLCNet_x0_75_train_infer_python.txt @@ -53,7 +53,7 @@ null:null batch_size:512 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/PPLCNet/PPLCNet_x1_0_train_infer_python.txt b/test_tipc/configs/PPLCNet/PPLCNet_x1_0_train_infer_python.txt index 6b941e682..4aa31ce7b 100644 --- a/test_tipc/configs/PPLCNet/PPLCNet_x1_0_train_infer_python.txt +++ b/test_tipc/configs/PPLCNet/PPLCNet_x1_0_train_infer_python.txt @@ -53,7 +53,7 @@ null:null batch_size:512 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/PPLCNet/PPLCNet_x2_5_train_infer_python.txt b/test_tipc/configs/PPLCNet/PPLCNet_x2_5_train_infer_python.txt index aee028991..717e67e56 100644 --- a/test_tipc/configs/PPLCNet/PPLCNet_x2_5_train_infer_python.txt +++ b/test_tipc/configs/PPLCNet/PPLCNet_x2_5_train_infer_python.txt @@ -50,10 +50,10 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o Global.benchmark:False null:null ===========================train_benchmark_params========================== -batch_size:256 +batch_size:512 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/PVTV2/PVT_V2_B2_Linear_train_infer_python.txt b/test_tipc/configs/PVTV2/PVT_V2_B2_Linear_train_infer_python.txt index e6097e132..f318f0138 100644 --- a/test_tipc/configs/PVTV2/PVT_V2_B2_Linear_train_infer_python.txt +++ b/test_tipc/configs/PVTV2/PVT_V2_B2_Linear_train_infer_python.txt @@ -52,9 +52,9 @@ null:null null:null ===========================train_benchmark_params========================== batch_size:128 -fp_items:fp32 +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/PVTV2/PVT_V2_B2_train_infer_python.txt b/test_tipc/configs/PVTV2/PVT_V2_B2_train_infer_python.txt index d607cfb8a..73e31e22d 100644 --- a/test_tipc/configs/PVTV2/PVT_V2_B2_train_infer_python.txt +++ b/test_tipc/configs/PVTV2/PVT_V2_B2_train_infer_python.txt @@ -50,5 +50,12 @@ 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|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]}] diff --git a/test_tipc/configs/SwinTransformer/SwinTransformer_base_patch4_window7_224_train_infer_python.txt b/test_tipc/configs/SwinTransformer/SwinTransformer_base_patch4_window7_224_train_infer_python.txt index 1ceef499a..daa661227 100644 --- a/test_tipc/configs/SwinTransformer/SwinTransformer_base_patch4_window7_224_train_infer_python.txt +++ b/test_tipc/configs/SwinTransformer/SwinTransformer_base_patch4_window7_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: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 +random_infer_input:[{float32,[3,224,224]}]