From 13ced08b8aeabc4f83772d5d5aef1ea4ccca3786 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Thu, 26 Aug 2021 09:40:32 +0000 Subject: [PATCH] add pretrained model to prune --- tests/ocr_det_params.txt | 2 +- tests/prepare.sh | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/ocr_det_params.txt b/tests/ocr_det_params.txt index 54a5583953..73b12cec9c 100644 --- a/tests/ocr_det_params.txt +++ b/tests/ocr_det_params.txt @@ -15,7 +15,7 @@ null:null trainer:norm_train|pact_train norm_train:tools/train.py -c configs/det/det_mv3_db.yml -o Global.pretrained_model=./pretrain_models/MobileNetV3_large_x0_5_pretrained pact_train:deploy/slim/quantization/quant.py -c configs/det/det_mv3_db.yml -o -fpgm_train:deploy/slim/prune/sensitivity_anal.py -c configs/det/det_mv3_db.yml -o +fpgm_train:deploy/slim/prune/sensitivity_anal.py -c configs/det/det_mv3_db.yml -o Global.pretrained_model=./pretrain_models/det_mv3_db_v2.0_train/best_accuracy distill_train:null null:null null:null diff --git a/tests/prepare.sh b/tests/prepare.sh index 0845df5d36..5da74d949f 100644 --- a/tests/prepare.sh +++ b/tests/prepare.sh @@ -34,6 +34,8 @@ MODE=$2 if [ ${MODE} = "lite_train_infer" ];then # pretrain lite train data wget -nc -P ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x0_5_pretrained.pdparams + wget -nc -P ./pretrain_models/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/en/det_mv3_db_v2.0_train.tar + cd ./pretrain_models/ && tar xf det_mv3_db_v2.0_train.tar && cd ../ rm -rf ./train_data/icdar2015 rm -rf ./train_data/ic15_data wget -nc -P ./train_data/ https://paddleocr.bj.bcebos.com/dygraph_v2.0/test/icdar2015_lite.tar