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

Expose noise_multiplier and l2_norm_clip as public read-only attributes.

Browse source 
PiperOrigin-RevId: 642298339
This commit is contained in:
William Kong 2024-06-11 09:38:13 -07:00 committed by A. Unique TensorFlower
parent 99c25e3be5
commit 00384db109
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tensorflow_privacy/privacy/keras_models/dp_keras_model.py
View file

@ -157,6 +157,14 @@ def make_dp_model_class(cls):
self.train_step, experimental_compile=True self.train_step, experimental_compile=True
) )
@property
def noise_multiplier(self):
return self._noise_multiplier
@property
def l2_norm_clip(self):
return self._l2_norm_clip
def _process_per_example_grads(self, grads): def _process_per_example_grads(self, grads):
grads_flat = tf.nest.flatten(grads) grads_flat = tf.nest.flatten(grads)
squared_l2_norms = [ squared_l2_norms = [