From 937cf5f047041cae59b6ee206e00eb86ad06ae59 Mon Sep 17 00:00:00 2001 From: Yu Yang Date: Thu, 16 Aug 2018 08:51:38 +0000 Subject: [PATCH] Fix bug --- fluid/neural_machine_translation/transformer/model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/fluid/neural_machine_translation/transformer/model.py b/fluid/neural_machine_translation/transformer/model.py index 5c11a5a3..bd83d758 100644 --- a/fluid/neural_machine_translation/transformer/model.py +++ b/fluid/neural_machine_translation/transformer/model.py @@ -522,13 +522,13 @@ def wrap_decoder(trg_vocab_size, x=dec_output, y=fluid.get_var(word_emb_param_names[0]), transpose_y=True) - predict = layers.softmax(predict) else: predict = layers.fc(input=dec_output, size=trg_vocab_size, bias_attr=False, - num_flatten_dims=2, - act='softmax') + num_flatten_dims=2) + if dec_inputs is None: + predict = layers.softmax(predict) return predict -- GitLab