Skip to content

M
models

PaddlePaddle / models
大约 1 年 前同步成功

通知 222
Star 6828
Fork 2962
M
models
未验证 提交 f573c2cf 编写于 作者: L liu zhengxi 提交者： GitHub
浏览文件
操作

Fix transformer save_inference_model (#4306)

上级 6423cab6
显示空白变更内容
内联 并排
Showing with 1 addition and 1 deletion
PaddleNLP/PaddleMT/transformer/inference_model.py
浏览文件 @ f573c2cf
...@@ -131,7 +131,7 @@ def do_save_inference_model(args): ...@@ -131,7 +131,7 @@ def do_save_inference_model(args):
fluid.io.save_inference_model( fluid.io.save_inference_model(
args.inference_model_dir, args.inference_model_dir,
feeded_var_names=input_field_names, feeded_var_names=list(input_field_names),
target_vars=[out_ids, out_scores], target_vars=[out_ids, out_scores],
executor=exe, executor=exe,
main_program=test_prog, main_program=test_prog,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或