From e66b53ca3e5772c1cd2ffa3c8852f3b1842a641b Mon Sep 17 00:00:00 2001 From: yukavio Date: Wed, 4 Nov 2020 05:23:36 +0000 Subject: [PATCH] fix some bug --- deploy/slim/prune/README.md | 2 +- deploy/slim/prune/pruning_and_finetune.py | 2 ++ deploy/slim/quantization/export_model.py | 3 ++- deploy/slim/quantization/quant.py | 1 + 4 files changed, 6 insertions(+), 2 deletions(-) diff --git a/deploy/slim/prune/README.md b/deploy/slim/prune/README.md index 20d8c1e92..9dc53f2cc 100644 --- a/deploy/slim/prune/README.md +++ b/deploy/slim/prune/README.md @@ -30,7 +30,7 @@ python setup.py install ``` ### 2. 获取预训练模型 -模型裁剪需要加载事先训练好的模型,PaddleOCR也提供了一系列模型[../../../doc/doc_ch/models_list.md],开发者可根据需要自行选择模型或使用自己的模型。 +模型裁剪需要加载事先训练好的模型,PaddleOCR也提供了一系列(模型)[../../../doc/doc_ch/models_list.md],开发者可根据需要自行选择模型或使用自己的模型。 ### 3. 敏感度分析训练 diff --git a/deploy/slim/prune/pruning_and_finetune.py b/deploy/slim/prune/pruning_and_finetune.py index 77ba604ad..24a115b7c 100644 --- a/deploy/slim/prune/pruning_and_finetune.py +++ b/deploy/slim/prune/pruning_and_finetune.py @@ -19,6 +19,7 @@ from __future__ import print_function import os import sys import numpy as np +import paddle __dir__ = os.path.dirname(__file__) sys.path.append(__dir__) sys.path.append(os.path.join(__dir__, '..', '..', '..')) @@ -49,6 +50,7 @@ skip_list = [ def main(): + paddle.enable_static() config = program.load_config(FLAGS.config) program.merge_config(FLAGS.opt) logger.info(config) diff --git a/deploy/slim/quantization/export_model.py b/deploy/slim/quantization/export_model.py index 5f4d91a8e..959c490d5 100644 --- a/deploy/slim/quantization/export_model.py +++ b/deploy/slim/quantization/export_model.py @@ -39,6 +39,7 @@ set_paddle_flags( ) import program +import paddle from paddle import fluid from ppocr.utils.utility import initial_logger logger = initial_logger() @@ -76,7 +77,7 @@ def main(): # The decay coefficient of moving average, default is 0.9 'moving_rate': 0.9, } - + paddle.enable_static() startup_prog, eval_program, place, config, alg_type = program.preprocess() feeded_var_names, target_vars, fetches_var_name = program.build_export( diff --git a/deploy/slim/quantization/quant.py b/deploy/slim/quantization/quant.py index e75e84be8..22e22a866 100755 --- a/deploy/slim/quantization/quant.py +++ b/deploy/slim/quantization/quant.py @@ -85,6 +85,7 @@ def get_optimizer(): def main(): + paddle.enable_static() train_build_outputs = program.build( config, train_program, startup_program, mode='train') train_loader = train_build_outputs[0]