We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0bd8e24 commit ceed1a3Copy full SHA for ceed1a3
transformer/cluttered_mnist.py
@@ -123,7 +123,7 @@
123
124
# %% Define loss/eval/training functions
125
cross_entropy = tf.reduce_mean(
126
- tf.nn.softmax_cross_entropy_with_logits(logits=y_logits, targets=y))
+ tf.nn.softmax_cross_entropy_with_logits(logits=y_logits, labels=y))
127
opt = tf.train.AdamOptimizer()
128
optimizer = opt.minimize(cross_entropy)
129
grads = opt.compute_gradients(cross_entropy, [b_fc_loc2])
0 commit comments