From 611f33240e451364dff78d342e6607fd0ce9df00 Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Tue, 5 Nov 2019 09:57:31 +0800 Subject: [PATCH] fix save inference model (#17) --- tools/export_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/export_model.py b/tools/export_model.py index b0c9edac3..4db937dd6 100644 --- a/tools/export_model.py +++ b/tools/export_model.py @@ -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) -- GitLab