From 86ee4a5782de105ee188a304efe0630a26a79ebe Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Wed, 19 Jan 2022 09:09:53 +0000
Subject: [PATCH] fix mode

---
 test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt | 4 ++--
 test_tipc/prepare.sh                                  | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt b/test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt
index d8829297..4f661d58 100644
--- a/test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt
+++ b/test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt
@@ -4,9 +4,9 @@ python:python3.7
 gpu_list:0|0,1
 Global.use_gpu:True|True
 Global.auto_cast:null
-Global.epoch_num:lite_train_lite_infer=2
+Global.epoch_num:benchmark_train=2
 Global.save_model_dir:./output/
-Train.loader.batch_size_per_card:lite_train_lite_infer=16
+Train.loader.batch_size_per_card:benchmark_train=16
 Global.pretrained_model:null
 train_model_name:latest
 train_infer_img_dir:null
diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
index 910ab5c5..2753aef4 100644
--- a/test_tipc/prepare.sh
+++ b/test_tipc/prepare.sh
@@ -21,6 +21,7 @@ model_name=$(func_parser_value "${lines[1]}")
 trainer_list=$(func_parser_value "${lines[14]}")
 
 if [ ${MODE} = "benchmark_train" ];then
+    pip install -r requirements.txt
     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
-- 
GitLab