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

Merge pull request #37 from ShadiRahimian:patch-1

Browse source 
PiperOrigin-RevId: 240601653
This commit is contained in:
A. Unique TensorFlower 2019-03-27 11:13:30 -07:00
commit a8e54d928c
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
privacy/analysis/compute_dp_sgd_privacy.py
View file

@ -80,8 +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] + 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))