minor fix encoder_decoding_sample (PaddlePaddle#934)

wangxicoding · Aug 27, 2021 · ac4dabf · ac4dabf
1 parent 047693d
commit ac4dabf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/paddlenlp/ops/faster_transformer/sample/encoder_decoding_sample.py b/paddlenlp/ops/faster_transformer/sample/encoder_decoding_sample.py
@@ -76,7 +76,8 @@ def do_predict(args):
         src_vocab_size=args.src_vocab_size,
         trg_vocab_size=args.trg_vocab_size,
         max_length=args.max_length + 1,
-        n_layer=args.n_layer,
+        num_encoder_layers=args.n_layer,
+        num_decoder_layers=args.n_layer,
         n_head=args.n_head,
         d_model=args.d_model,
         d_inner_hid=args.d_inner_hid,