626_privacy/tensorflow_privacy
2
0
Fork 1
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

update keras model

Browse source
This commit is contained in:
pranav subramani 2021-01-08 00:23:32 -07:00
parent 6982e027b5
commit 13b3a04a3e
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/keras_models/dp_keras_model.py
View file

@ -27,7 +27,7 @@ def make_dp_model_class(cls):
noise = tf.random.normal(
tf.shape(input=summed_grads), stddev=noise_stddev)
noised_grads = summed_grads + noise
return noised_grads / tf.cast(stacked_grads.shape[0], tf.float32)
return noised_grads / tf.cast(stacked_grads.shape[0], noised_grads.dtype)
def compute_per_example_grads(self, data):
x, y = data