Skip to content

Commit

Permalink
Merge pull request #2 from metropt/fix_arguments_tf1
Browse files Browse the repository at this point in the history
named arguments on function sparse_softmax_cross_entropy_with_logits
  • Loading branch information
Wolfgang Beyer committed Mar 23, 2017
2 parents e0c769d + 333bcfc commit 9a319f2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions softmax.py
Expand Up @@ -40,8 +40,8 @@
logits = tf.matmul(images_placeholder, weights) + biases

# Define the loss function
loss = tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits(logits,
labels_placeholder))
loss = tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits(logits=logits,
labels=labels_placeholder))

# Define the training operation
train_step = tf.train.GradientDescentOptimizer(learning_rate).minimize(loss)
Expand Down

0 comments on commit 9a319f2

Please sign in to comment.