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

Update compute_dp_sgd_privacy.py

Browse source
This commit is contained in:
Nicolas Papernot 2019-03-27 09:18:45 -07:00 committed by GitHub
parent 0a0e5cb3c3
commit 8db2dd6bca
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
privacy/analysis/compute_dp_sgd_privacy.py
View file

@ -80,7 +80,8 @@ def main(argv):
if q > 1:
raise app.UsageError('N must be larger than the batch size.')
orders = [1.25, 1.5, 1.75, 2., 2.25, 2.5, 3., 3.5, 4., 4.5] + list(range(5, 64)) + [128, 256, 512]
orders = ([1.25, 1.5, 1.75, 2., 2.25, 2.5, 3., 3.5, 4., 4.5] +
list(range(5, 64)) + [128, 256, 512])
steps = int(math.ceil(FLAGS.epochs * FLAGS.N / FLAGS.batch_size))