Update compute_dp_sgd_privacy.py
This commit is contained in:
parent
8db2dd6bca
commit
01e7cac7b5
1 changed files with 1 additions and 1 deletions
|
@ -80,7 +80,7 @@ def main(argv):
|
||||||
if q > 1:
|
if q > 1:
|
||||||
raise app.UsageError('N must be larger than the batch size.')
|
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] +
|
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])
|
list(range(5, 64)) + [128, 256, 512])
|
||||||
|
|
||||||
steps = int(math.ceil(FLAGS.epochs * FLAGS.N / FLAGS.batch_size))
|
steps = int(math.ceil(FLAGS.epochs * FLAGS.N / FLAGS.batch_size))
|
||||||
|
|
Loading…
Reference in a new issue