From 65d356b09bab422c1b31d289d4ad7617cf139017 Mon Sep 17 00:00:00 2001 From: Ryan McAvoy Date: Wed, 18 Apr 2018 09:32:51 -0700 Subject: [PATCH] fixed softmax cross entropy with logits to use v2 --- tensor2tensor/layers/common_layers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensor2tensor/layers/common_layers.py b/tensor2tensor/layers/common_layers.py index 3c1155643..2f8443c66 100644 --- a/tensor2tensor/layers/common_layers.py +++ b/tensor2tensor/layers/common_layers.py @@ -1824,7 +1824,7 @@ def smoothing_cross_entropy(logits, depth=vocab_size, on_value=confidence, off_value=low_confidence) - xentropy = tf.nn.softmax_cross_entropy_with_logits( + xentropy = tf.nn.softmax_cross_entropy_with_logits_v2( logits=logits, labels=soft_targets) return xentropy - normalizing