From 36bf71243cb59e077204b4d323aba92ec04bc07b Mon Sep 17 00:00:00 2001
From: WenmuZhou <572459439@qq.com>
Date: Tue, 6 Sep 2022 07:16:29 +0000
Subject: [PATCH] add pretrain model for slanet in benckmark

---
 test_tipc/benchmark_train.sh | 13 +++++++------
 test_tipc/prepare.sh         |  2 ++
 2 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/test_tipc/benchmark_train.sh b/test_tipc/benchmark_train.sh
index 1dcb0129..25fda8f9 100644
--- a/test_tipc/benchmark_train.sh
+++ b/test_tipc/benchmark_train.sh
@@ -1,12 +1,6 @@
 #!/bin/bash
 source test_tipc/common_func.sh
 
-# set env
-python=python
-export str_tmp=$(echo `pip list|grep paddlepaddle-gpu|awk -F ' ' '{print $2}'`)
-export frame_version=${str_tmp%%.post*}
-export frame_commit=$(echo `${python} -c "import paddle;print(paddle.version.commit)"`)
-
 # run benchmark sh 
 # Usage:
 # bash run_benchmark_train.sh config.txt params
@@ -86,6 +80,13 @@ dataline=`cat $FILENAME`
 IFS=$'\n'
 lines=(${dataline})
 model_name=$(func_parser_value "${lines[1]}")
+python_name=$(func_parser_value "${lines[2]}")
+
+# set env
+python=${python_name}
+export str_tmp=$(echo `pip list|grep paddlepaddle-gpu|awk -F ' ' '{print $2}'`)
+export frame_version=${str_tmp%%.post*}
+export frame_commit=$(echo `${python} -c "import paddle;print(paddle.version.commit)"`)
 
 # 获取benchmark_params所在的行数
 line_num=`grep -n "train_benchmark_params" $FILENAME  | cut -d ":" -f 1`
diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
index 388dad0d..5d50a5ad 100644
--- a/test_tipc/prepare.sh
+++ b/test_tipc/prepare.sh
@@ -108,6 +108,8 @@ if [ ${MODE} = "benchmark_train" ];then
         cd ../
     fi
     if [[ ${model_name} == "slanet" ]];then
+        wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/ppstructure/models/slanet/en_ppstructure_mobile_v2.0_SLANet_train.tar --no-check-certificate
+        cd ./pretrain_models/ && tar xf en_ppstructure_mobile_v2.0_SLANet_train.tar  && cd ../
         rm -rf ./train_data/pubtabnet
         wget -nc -P ./train_data/ https://paddleocr.bj.bcebos.com/dataset/pubtabnet_benckmark.tar --no-check-certificate
         cd ./train_data/ && tar xf pubtabnet_benckmark.tar
-- 
GitLab