未验证 提交 f573c2cf 编写于 作者: L liu zhengxi 提交者: GitHub

Fix transformer save_inference_model (#4306)

上级 6423cab6
......@@ -131,7 +131,7 @@ def do_save_inference_model(args):
fluid.io.save_inference_model(
args.inference_model_dir,
feeded_var_names=input_field_names,
feeded_var_names=list(input_field_names),
target_vars=[out_ids, out_scores],
executor=exe,
main_program=test_prog,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册