From 777857ec8944b180788da9f888bdff81ebbe1f78 Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Thu, 17 Jun 2021 12:02:05 +0800
Subject: [PATCH] rename quant and prune

---
 test/infer.sh                | 4 ++--
 test/paddleocr_ci_params.txt | 4 ++--
 test/test.sh                 | 5 +++--
 3 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/test/infer.sh b/test/infer.sh
index eb70cafb..78057705 100644
--- a/test/infer.sh
+++ b/test/infer.sh
@@ -76,7 +76,7 @@ for train_model in ${train_model_list[*]}; do
                 wget -nc  -P ./inference https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_rec_train.tar
                 cd ./inference && tar xf ${eval_model_name}.tar && cd ../
             fi 
-        elif [ ${slim_trainer} = "quant" ]; then
+        elif [ ${slim_trainer} = "pact" ]; then
             if [ ${model_name} = "det" ]; then
                 eval_model_name="ch_ppocr_mobile_v2.0_det_quant_train"
                 wget -nc  -P ./inference https://paddleocr.bj.bcebos.com/dygraph_v2.0/slim/ch_ppocr_mobile_v2.0_det_quant_train.tar
@@ -96,7 +96,7 @@ for train_model in ${train_model_list[*]}; do
                 wget -nc  -P ./inference https://paddleocr.bj.bcebos.com/dygraph_v2.0/slim/ch_ppocr_mobile_v2.0_rec_distill_train.tar
                 cd ./inference && tar xf ${eval_model_name}.tar && cd ../
             fi 
-        elif [ ${slim_trainer} = "prune" ]; then
+        elif [ ${slim_trainer} = "fpgm" ]; then
             if [ ${model_name} = "det" ]; then
                 eval_model_name="ch_ppocr_mobile_v2.0_det_prune_train"
                 wget -nc  -P ./inference https://paddleocr.bj.bcebos.com/dygraph_v2.0/slim/ch_ppocr_mobile_v2.0_det_prune_train.tar
diff --git a/test/paddleocr_ci_params.txt b/test/paddleocr_ci_params.txt
index 93c5ccde..4cd035ea 100644
--- a/test/paddleocr_ci_params.txt
+++ b/test/paddleocr_ci_params.txt
@@ -1,7 +1,7 @@
 train_model_list: ocr_det
 gpu_list: -1|0|0,1
-auto_cast_list: False
-trainer_list: norm|quant
+auto_cast_list: False|True
+trainer_list: norm|pact|fpgm
 python: python3.7
 
 inference: python
diff --git a/test/test.sh b/test/test.sh
index 92c61f43..02379baa 100644
--- a/test/test.sh
+++ b/test/test.sh
@@ -75,6 +75,7 @@ ${python} -m pip install psutil;
 ${python} -m pip install GPUtil;
 ${python} -m pip install paddlesim==2.0.0
 
+
 function status_check(){
     last_status=$1   # the exit code
     run_model=$2
@@ -119,11 +120,11 @@ for train_model in ${train_model_list[*]}; do
                     trainer="tools/train.py"
                     export_model="tools/export_model.py"
                     pretrain="./pretrain_models/MobileNetV3_large_x0_5_pretrained"
-                elif [ ${slim_trainer} = "quant" ]; then
+                elif [ ${slim_trainer} = "pact" ]; then
                     trainer="deploy/slim/quantization/quant.py"
                     export_model="deploy/slim/quantization/export_model.py"
                     pretrain="./pretrain_models/det_mv3_db_v2.0_train/best_accuracy"
-                elif [ ${slim_trainer} = "prune" ]; then
+                elif [ ${slim_trainer} = "fpgm" ]; then
                     trainer="deploy/slim/prune/sensitivity_anal.py"
                     export_model="deploy/slim/prune/export_prune_model.py"
                     pretrain="./pretrain_models/det_mv3_db_v2.0_train/best_accuracy"
-- 
GitLab