We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6918848 commit 7cc5f2fCopy full SHA for 7cc5f2f
tensorlayer/cost.py
@@ -36,7 +36,7 @@ def cross_entropy(output, target, name=None):
36
if tf.__version__ <= "0.12":
37
return tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits(logits=output, targets=target, name=name))
38
else: # TF 1.0
39
- return tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits(labels=target, logits=outputs, name=name))
+ return tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits(labels=target, logits=output, name=name))
40
41
42
0 commit comments