Update cluttered_mnist.py

martinwicke · web-flow · commit bda41af1e96b · 2017-01-08T01:21:32.000-08:00
diff --git a/transformer/cluttered_mnist.py b/transformer/cluttered_mnist.py
@@ -123,7 +123,7 @@
 
 # %% Define loss/eval/training functions
 cross_entropy = tf.reduce_mean(
-    tf.nn.softmax_cross_entropy_with_logits(y_logits, y))
+    tf.nn.softmax_cross_entropy_with_logits(logits=y_logits, targets=y))
 opt = tf.train.AdamOptimizer()
 optimizer = opt.minimize(cross_entropy)
 grads = opt.compute_gradients(cross_entropy, [b_fc_loc2])
