未验证 提交 14fad5d0 编写于 作者: 三毛的二哥's avatar 三毛的二哥 提交者: GitHub

ocr_recognition的train.py更新:断点训练加载模型时缺少参数 “exe”

ocr_recognition的train.py代码(release 1.7)第109行fluid.load()缺少第三个参数"exe",正确的写法如下:
    # load init model
    if args.init_model is not None:
        model_dir = args.init_model
        fluid.load(fluid.default_main_program(), model_dir, exe)
        print("Init model from: %s." % args.init_model)
若不加“exe”参数,加载已训练的模型进行断点训练时将会有如下报错:
AttributeError: 'NoneType' object has no attribute 'get_tensor'
上级 08316fb8
......@@ -106,7 +106,7 @@ def train(args):
# load init model
if args.init_model is not None:
model_dir = args.init_model
fluid.load(fluid.default_main_program(), model_dir)
fluid.load(fluid.default_main_program(), model_dir, exe)
print("Init model from: %s." % args.init_model)
train_exe = exe
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册