Forráskód Böngészése

Update deep_cnn.py

Martin Wicke 8 éve
szülő
commit
9e9aa5d1dc
1 módosított fájl, 1 hozzáadás és 1 törlés
  1. 1 1
      differential_privacy/multiple_teachers/deep_cnn.py

+ 1 - 1
differential_privacy/multiple_teachers/deep_cnn.py

@@ -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')