Skip to content

Commit 319bf81

Browse files
Setting correct argument in super() call in DPOptimizerClass.
PiperOrigin-RevId: 308776729
1 parent 9259ccb commit 319bf81

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow_privacy/privacy/optimizers/dp_optimizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ def normalize(v):
135135

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

0 commit comments

Comments
 (0)