From 9c7c5a53ee42dfcb83644bda502b96e7ce0af7b3 Mon Sep 17 00:00:00 2001
From: liu zhengxi <380185688@qq.com>
Date: Mon, 7 Dec 2020 17:12:42 +0800
Subject: [PATCH] fix predict (#4985)

---
 PaddleNLP/examples/machine_translation/transformer/reader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/PaddleNLP/examples/machine_translation/transformer/reader.py b/PaddleNLP/examples/machine_translation/transformer/reader.py
index 19134b36..b29db4c0 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__)
-- 
GitLab