未验证 提交 450160e8 编写于 作者: 0 0YuanZhang0 提交者: GitHub

upgrade ade dgu load api (#4659)

* fix_ade_readme

* fix_load_api
上级 b7ae38a6
...@@ -71,9 +71,9 @@ def do_save_inference_model(args): ...@@ -71,9 +71,9 @@ def do_save_inference_model(args):
assert (args.init_from_params) or (args.init_from_pretrain_model) assert (args.init_from_params) or (args.init_from_pretrain_model)
if args.init_from_params: if args.init_from_params:
fluid.load(test_prog, args.init_from_params) fluid.load(test_prog, args.init_from_params, executor=exe)
elif args.init_from_pretrain_model: elif args.init_from_pretrain_model:
fluid.load(test_prog, args.init_from_pretrain_model) fluid.load(test_prog, args.init_from_pretrain_model, executor=exe)
# saving inference model # saving inference model
fluid.io.save_inference_model( fluid.io.save_inference_model(
......
...@@ -99,7 +99,7 @@ def do_save_inference_model(args): ...@@ -99,7 +99,7 @@ def do_save_inference_model(args):
assert (args.init_from_params) assert (args.init_from_params)
if args.init_from_params: if args.init_from_params:
fluid.load(test_prog, args.init_from_params) fluid.load(test_prog, args.init_from_params, executor=exe)
# saving inference model # saving inference model
fluid.io.save_inference_model( fluid.io.save_inference_model(
......
...@@ -109,7 +109,7 @@ def do_predict(args): ...@@ -109,7 +109,7 @@ def do_predict(args):
assert (args.init_from_params) assert (args.init_from_params)
if args.init_from_params: if args.init_from_params:
fluid.load(test_prog, args.init_from_params) fluid.load(test_prog, args.init_from_params, executor=exe)
compiled_test_prog = fluid.CompiledProgram(test_prog) compiled_test_prog = fluid.CompiledProgram(test_prog)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册