diff options
| author |  Manjunath Kudlur <keveman@gmail.com> | 2016-02-05 08:45:20 -0800 |
|---|---|---|
| committer | Manjunath Kudlur <keveman@gmail.com> | 2016-02-05 08:47:08 -0800 |
| commit | 434b79ec286c13bc0fb8b65ff824cf7ea3ee0c3e (patch) | |
| tree | 29eb1fbcb8416f23364b06c37667d8911edab43c | |
| parent | 16e91ab965d54ede8ceda946cd1db125a9901ec5 (diff) | |
Remove vocab_size argument for model_with_buckets.
Change: 113952458
| -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) |