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

Passes number of microbatches to DP model.

Browse source 
PiperOrigin-RevId: 512678620
This commit is contained in:
Shuang Song 2023-02-27 11:11:27 -08:00 committed by A. Unique TensorFlower
parent 4a418e8862
commit a3e8a45559
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tutorials/mnist_dpsgd_tutorial_keras_model.py
View file

@ -113,7 +113,9 @@ def main(unused_argv):
model = DPSequential(
l2_norm_clip=FLAGS.l2_norm_clip,
noise_multiplier=FLAGS.noise_multiplier,
layers=layers)
num_microbatches=FLAGS.microbatches,
layers=layers,
)
else:
model = tf.keras.Sequential(layers=layers)