소스 검색

Fix xent call in mnist tutorial code

Fixes #857.
Martin Wicke 8 년 전
부모
커밋
e93ec37201
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      tutorials/image/mnist/convolutional.py

+ 1 - 1
tutorials/image/mnist/convolutional.py

@@ -228,7 +228,7 @@ def main(_):
   # Training computation: logits + cross-entropy loss.
   logits = model(train_data_node, True)
   loss = tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits(
-      logits, train_labels_node))
+      labels=train_labels_node, logits=logits))
 
   # L2 regularization for the fully connected parameters.
   regularizers = (tf.nn.l2_loss(fc1_weights) + tf.nn.l2_loss(fc1_biases) +