From d132f4e8db74f0a00e62c493e0b3a9ea106f3944 Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Fri, 1 Apr 2022 14:23:45 +0000 Subject: [PATCH] Add MobileViT train bencnmark --- .../config/MobileViT/MobileViT_S_train_infer_python.txt | 6 ++++++ .../config/MobileViT/MobileViT_XS_train_infer_python.txt | 6 ++++++ .../config/MobileViT/MobileViT_XXS_train_infer_python.txt | 6 ++++++ 3 files changed, 18 insertions(+) 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 950f3ee4..d45bbb99 100644 --- a/test_tipc/config/MobileViT/MobileViT_S_train_infer_python.txt +++ b/test_tipc/config/MobileViT/MobileViT_S_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 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 826bd4f8..a366d42c 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,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 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 83902b67..cc716941 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,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