From b2d80a7a53ecd5403b35ebf88644c4ab9206d83c Mon Sep 17 00:00:00 2001
From: hysunflower <nothing_upc@163.com>
Date: Tue, 23 Nov 2021 12:13:07 +0000
Subject: [PATCH] update_scripts_benchmark

---
 benchmark/run_all.sh | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/benchmark/run_all.sh b/benchmark/run_all.sh
index 648d89783..3c6cee83d 100644
--- a/benchmark/run_all.sh
+++ b/benchmark/run_all.sh
@@ -8,11 +8,13 @@
 model_mode_list=(MobileNetV1 MobileNetV2 MobileNetV3_large_x1_0 ShuffleNetV2_x1_0 HRNet_W48_C SwinTransformer_tiny_patch4_window7_224 alt_gvt_base)    # benchmark 监控模型列表
 #model_mode_list=(MobileNetV1 MobileNetV2 MobileNetV3_large_x1_0 EfficientNetB0 ShuffleNetV2_x1_0 DenseNet121 HRNet_W48_C SwinTransformer_tiny_patch4_window7_224 alt_gvt_base)   # 该脚本支持列表
 fp_item_list=(fp32)
-bs_list=(64)     #32 64 96 128)
+#bs_list=(32 64 96 128)
 for model_mode in ${model_mode_list[@]}; do
       for fp_item in ${fp_item_list[@]}; do
           if [ ${model_mode} = MobileNetV3_large_x1_0 ] || [ ${model_mode} = ShuffleNetV2_x1_0 ]; then
               bs_list=(256)
+          else
+              bs_list=(64)
           fi
           for bs_item in ${bs_list[@]};do
 	    echo "index is speed, 1gpus, begin, ${model_name}"