diff options
-rw-r--r-- | tensorflow/models/rnn/translate/seq2seq_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/models/rnn/translate/seq2seq_model.py b/tensorflow/models/rnn/translate/seq2seq_model.py index 3a5f8b8b60..3b4c89f5c2 100644 --- a/tensorflow/models/rnn/translate/seq2seq_model.py +++ b/tensorflow/models/rnn/translate/seq2seq_model.py @@ -147,7 +147,7 @@ class Seq2SeqModel(object): else: self.outputs, self.losses = seq2seq.model_with_buckets( self.encoder_inputs, self.decoder_inputs, targets, - self.target_weights, buckets, self.target_vocab_size, + self.target_weights, buckets, lambda x, y: seq2seq_f(x, y, False), softmax_loss_function=softmax_loss_function) |