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"