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 1906b16 commit f24919eCopy full SHA for f24919e
LeNet-Lab.ipynb
@@ -287,7 +287,7 @@
287
"rate = 0.001\n",
288
"\n",
289
"logits = LeNet(x)\n",
290
- "cross_entropy = tf.nn.softmax_cross_entropy_with_logits(logits, one_hot_y)\n",
+ "cross_entropy = tf.nn.softmax_cross_entropy_with_logits(labels=one_hot_y, logits=logits)\n",
291
"loss_operation = tf.reduce_mean(cross_entropy)\n",
292
"optimizer = tf.train.AdamOptimizer(learning_rate = rate)\n",
293
"training_operation = optimizer.minimize(loss_operation)"
0 commit comments