diff --git a/PaddleNLP/examples/machine_translation/transformer/reader.py b/PaddleNLP/examples/machine_translation/transformer/reader.py index 19134b36562766c3364ce7f98a56386ec180ac21..b29db4c082466cdb92244fd04f37bd7795856ab6 100644 --- a/PaddleNLP/examples/machine_translation/transformer/reader.py +++ b/PaddleNLP/examples/machine_translation/transformer/reader.py @@ -49,7 +49,7 @@ def create_infer_loader(args): prepare_infer_input, bos_idx=args.bos_idx, eos_idx=args.eos_idx, - pad_idx=args.eos_idx), + pad_idx=args.bos_idx), num_workers=0, return_list=True) data_loaders = (data_loader, batch_sampler.__len__)