Skip to content

Commit ceed1a3

Browse files
Asiapenolovenealwu
authored andcommitted
old argument targets->labels of tf.nn.softmax_cross_entropy_with_logits for tf 1.0
1 parent 0bd8e24 commit ceed1a3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

transformer/cluttered_mnist.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@
123123

124124
# %% Define loss/eval/training functions
125125
cross_entropy = tf.reduce_mean(
126-
tf.nn.softmax_cross_entropy_with_logits(logits=y_logits, targets=y))
126+
tf.nn.softmax_cross_entropy_with_logits(logits=y_logits, labels=y))
127127
opt = tf.train.AdamOptimizer()
128128
optimizer = opt.minimize(cross_entropy)
129129
grads = opt.compute_gradients(cross_entropy, [b_fc_loc2])

0 commit comments

Comments
 (0)