提交 e66b53ca 编写于 作者: Y yukavio

fix some bug

上级 eb386cdb
...@@ -30,7 +30,7 @@ python setup.py install ...@@ -30,7 +30,7 @@ python setup.py install
``` ```
### 2. 获取预训练模型 ### 2. 获取预训练模型
模型裁剪需要加载事先训练好的模型,PaddleOCR也提供了一系列模型[../../../doc/doc_ch/models_list.md],开发者可根据需要自行选择模型或使用自己的模型。 模型裁剪需要加载事先训练好的模型,PaddleOCR也提供了一系列(模型)[../../../doc/doc_ch/models_list.md],开发者可根据需要自行选择模型或使用自己的模型。
### 3. 敏感度分析训练 ### 3. 敏感度分析训练
......
...@@ -19,6 +19,7 @@ from __future__ import print_function ...@@ -19,6 +19,7 @@ from __future__ import print_function
import os import os
import sys import sys
import numpy as np import numpy as np
import paddle
__dir__ = os.path.dirname(__file__) __dir__ = os.path.dirname(__file__)
sys.path.append(__dir__) sys.path.append(__dir__)
sys.path.append(os.path.join(__dir__, '..', '..', '..')) sys.path.append(os.path.join(__dir__, '..', '..', '..'))
...@@ -49,6 +50,7 @@ skip_list = [ ...@@ -49,6 +50,7 @@ skip_list = [
def main(): def main():
paddle.enable_static()
config = program.load_config(FLAGS.config) config = program.load_config(FLAGS.config)
program.merge_config(FLAGS.opt) program.merge_config(FLAGS.opt)
logger.info(config) logger.info(config)
......
...@@ -39,6 +39,7 @@ set_paddle_flags( ...@@ -39,6 +39,7 @@ set_paddle_flags(
) )
import program import program
import paddle
from paddle import fluid from paddle import fluid
from ppocr.utils.utility import initial_logger from ppocr.utils.utility import initial_logger
logger = initial_logger() logger = initial_logger()
...@@ -76,7 +77,7 @@ def main(): ...@@ -76,7 +77,7 @@ def main():
# The decay coefficient of moving average, default is 0.9 # The decay coefficient of moving average, default is 0.9
'moving_rate': 0.9, 'moving_rate': 0.9,
} }
paddle.enable_static()
startup_prog, eval_program, place, config, alg_type = program.preprocess() startup_prog, eval_program, place, config, alg_type = program.preprocess()
feeded_var_names, target_vars, fetches_var_name = program.build_export( feeded_var_names, target_vars, fetches_var_name = program.build_export(
......
...@@ -85,6 +85,7 @@ def get_optimizer(): ...@@ -85,6 +85,7 @@ def get_optimizer():
def main(): def main():
paddle.enable_static()
train_build_outputs = program.build( train_build_outputs = program.build(
config, train_program, startup_program, mode='train') config, train_program, startup_program, mode='train')
train_loader = train_build_outputs[0] train_loader = train_build_outputs[0]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册