aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/legacy_seq2seq
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2016-12-06 13:21:11 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-12-06 16:01:32 -0800
commit24246a1097048883bc83951cda347c447c562bcc (patch)
tree9736a1ae188dbd3c15b284f17ec99b93560c9076 /tensorflow/contrib/legacy_seq2seq
parent8fbde6332580bb05a0f510cfc71602b87f39b275 (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.py2
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(