未验证 提交 789012c0 编写于 作者: 李季 提交者: GitHub

fix the undefined variable bug in dist_transformer file (#36211) (#36233)

上级 e8efba57
...@@ -1450,7 +1450,7 @@ def wrap_decoder(trg_vocab_size, ...@@ -1450,7 +1450,7 @@ def wrap_decoder(trg_vocab_size,
# This is used to implement independent decoder program in inference. # This is used to implement independent decoder program in inference.
trg_word, trg_pos, trg_slf_attn_bias, trg_src_attn_bias, \ trg_word, trg_pos, trg_slf_attn_bias, trg_src_attn_bias, \
enc_output = make_all_inputs( enc_output = make_all_inputs(
decoder_data_input_fields + decoder_util_input_fields) decoder_data_input_fields)
else: else:
trg_word, trg_pos, trg_slf_attn_bias, trg_src_attn_bias = dec_inputs trg_word, trg_pos, trg_slf_attn_bias, trg_src_attn_bias = dec_inputs
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册