diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2016-12-06 13:21:11 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-12-06 16:01:32 -0800 |
commit | 24246a1097048883bc83951cda347c447c562bcc (patch) | |
tree | 9736a1ae188dbd3c15b284f17ec99b93560c9076 /tensorflow/contrib/legacy_seq2seq | |
parent | 8fbde6332580bb05a0f510cfc71602b87f39b275 (diff) |
Swaps the expected order of softmax_loss(labels, inputs) arguments for seq2seq loss functions.
Change: 141217254
Diffstat (limited to 'tensorflow/contrib/legacy_seq2seq')
-rw-r--r-- | tensorflow/contrib/legacy_seq2seq/python/kernel_tests/seq2seq_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/legacy_seq2seq/python/kernel_tests/seq2seq_test.py b/tensorflow/contrib/legacy_seq2seq/python/kernel_tests/seq2seq_test.py index 1996f127c1..33b5b96ca8 100644 --- a/tensorflow/contrib/legacy_seq2seq/python/kernel_tests/seq2seq_test.py +++ b/tensorflow/contrib/legacy_seq2seq/python/kernel_tests/seq2seq_test.py @@ -608,7 +608,7 @@ class Seq2SeqTest(tf.test.TestCase): num_decoder_symbols=classes, embedding_size=24, output_projection=(w, b)) targets = [dec_inp[i+1] for i in range(len(dec_inp) - 1)] + [0] - def SampledLoss(inputs, labels): + def SampledLoss(labels, inputs): labels = tf.reshape(labels, [-1, 1]) return tf.nn.sampled_softmax_loss(w_t, b, inputs, labels, 8, classes) return tf.contrib.legacy_seq2seq.model_with_buckets( |