未验证 提交 611f3324 编写于 作者: W wangguanzhong 提交者: GitHub

fix save inference model (#17)

上级 1f5b8d5c
......@@ -97,7 +97,7 @@ def main():
infer_prog = fluid.Program()
with fluid.program_guard(infer_prog, startup_prog):
with fluid.unique_name.guard():
_, feed_vars = create_feed(test_feed, use_pyreader=False)
_, feed_vars = create_feed(test_feed, iterable=True)
test_fetches = model.test(feed_vars)
infer_prog = infer_prog.clone(True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册