From f32f31bf76b854189545ae6ea5e544e270fa22a0 Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Wed, 19 Jan 2022 06:44:02 +0000
Subject: [PATCH] mode to benchmark_train

---
 test_tipc/prepare.sh | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
index 701a91c5..910ab5c5 100644
--- a/test_tipc/prepare.sh
+++ b/test_tipc/prepare.sh
@@ -20,6 +20,14 @@ model_name=$(func_parser_value "${lines[1]}")
 
 trainer_list=$(func_parser_value "${lines[14]}")
 
+if [ ${MODE} = "benchmark_train" ];then
+    if [[ ${model_name} =~ "det_mv3_db_v2.0_benchmark" ]];then
+        rm -rf ./train_data/icdar2015
+        wget -nc -P  ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x0_5_pretrained.pdparams  --no-check-certificate
+        wget -nc -P ./train_data/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/icdar2015.tar --no-check-certificate
+        cd ./train_data/ && tar xf icdar2015.tar && cd ../
+    fi
+fi
 
 if [ ${MODE} = "lite_train_lite_infer" ];then
     # pretrain lite train data
@@ -65,11 +73,6 @@ if [ ${MODE} = "lite_train_lite_infer" ];then
         wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_rec_train.tar --no-check-certificate
         cd ./pretrain_models/ && tar xf ch_ppocr_mobile_v2.0_rec_train.tar && cd ../
     fi
-    if [[ ${model_name} =~ "det_mv3_db_v2.0_benchmark" ]];then
-        rm -rf ./train_data/icdar2015
-        wget -nc -P ./train_data/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/icdar2015.tar --no-check-certificate
-        cd ./train_data/ && tar xf icdar2015.tar && cd ../
-    fi
 
 elif [ ${MODE} = "whole_train_whole_infer" ];then
     wget -nc -P  ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x0_5_pretrained.pdparams --no-check-certificate
-- 
GitLab