Skip to content

Commit

Permalink
Update deep_cnn.py
Browse files Browse the repository at this point in the history
  • Loading branch information
martinwicke authored Jan 8, 2017
1 parent 84a32a8 commit 9e9aa5d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion differential_privacy/multiple_teachers/deep_cnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -341,7 +341,7 @@ def loss_fun(logits, labels):
# Calculate the cross entropy between labels and predictions
labels = tf.cast(labels, tf.int64)
cross_entropy = tf.nn.sparse_softmax_cross_entropy_with_logits(
logits, labels, name='cross_entropy_per_example')
logits=logits, labels=labels, name='cross_entropy_per_example')

# Calculate the average cross entropy loss across the batch.
cross_entropy_mean = tf.reduce_mean(cross_entropy, name='cross_entropy')
Expand Down

0 comments on commit 9e9aa5d

Please sign in to comment.