ruby/tensorflow_privacy
1
0
Fork 0
forked from 626_privacy/tensorflow_privacy
Code Pull requests Activity

Setting correct argument in super() call in DPOptimizerClass.

Browse source 
PiperOrigin-RevId: 308776729
This commit is contained in:
A. Unique TensorFlower 2020-04-28 00:37:35 -07:00
parent 9259ccb3d8
commit 319bf81bb1
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tensorflow_privacy/privacy/optimizers/dp_optimizer.py
View file

@ -135,7 +135,7 @@ def make_optimizer_class(cls):
def process_microbatch(i, sample_state): def process_microbatch(i, sample_state):
"""Process one microbatch (record) with privacy helper.""" """Process one microbatch (record) with privacy helper."""
grads, _ = zip(*super(cls, self).compute_gradients( grads, _ = zip(*super(DPOptimizerClass, self).compute_gradients(
tf.reduce_mean(input_tensor=tf.gather( tf.reduce_mean(input_tensor=tf.gather(
microbatches_losses, [i])), var_list, gate_gradients, microbatches_losses, [i])), var_list, gate_gradients,
aggregation_method, colocate_gradients_with_ops, grad_loss)) aggregation_method, colocate_gradients_with_ops, grad_loss))