Skip to content

M
models

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

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

fix predict (#4985)

上级 755e53f7
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
PaddleNLP/examples/machine_translation/transformer/reader.py
浏览文件 @ 9c7c5a53
...@@ -49,7 +49,7 @@ def create_infer_loader(args): ...@@ -49,7 +49,7 @@ def create_infer_loader(args):
prepare_infer_input, prepare_infer_input,
bos_idx=args.bos_idx, bos_idx=args.bos_idx,
eos_idx=args.eos_idx, eos_idx=args.eos_idx,
pad_idx=args.eos_idx), pad_idx=args.bos_idx),
num_workers=0, num_workers=0,
return_list=True) return_list=True)
data_loaders = (data_loader, batch_sampler.__len__) data_loaders = (data_loader, batch_sampler.__len__)
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或