add pretrained model to prune

truongnn1106 · Aug 26, 2021 · 13ced08 · 13ced08
1 parent 58ef7ac
commit 13ced08
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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