From 4f1b432a1b3d8e2bed0cc999648059edb806cf2c Mon Sep 17 00:00:00 2001
From: hysunflower <nothing_upc@163.com>
Date: Tue, 30 Nov 2021 07:15:11 +0000
Subject: [PATCH] fix_log_name

---
 benchmark/run_det.sh | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/benchmark/run_det.sh b/benchmark/run_det.sh
index 4cf32f77..4f0e7142 100644
--- a/benchmark/run_det.sh
+++ b/benchmark/run_det.sh
@@ -13,9 +13,13 @@ wget -P ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dyg
 
 model_mode_list=(det_res18_db_v2.0 det_r50_vd_east det_r50_vd_pse)
 fp_item_list=(fp32)
-bs_list=(8 16)
 for model_mode in ${model_mode_list[@]}; do
       for fp_item in ${fp_item_list[@]}; do
+          if [ ${model_mode} == "det_res18_db_v2.0" ]; then
+              bs_list=(8 16)
+          else
+              bs_list=(16)
+          fi
           for bs_item in ${bs_list[@]}; do
             log_name=ocr_${model_mode}_bs${bs_item}_${fp_item}
             echo "index is speed, 1gpus, begin, ${model_name}"
-- 
GitLab