From 8db2dd6bca656beafe7f29e5711d37959d71d2ce Mon Sep 17 00:00:00 2001 From: Nicolas Papernot Date: Wed, 27 Mar 2019 09:18:45 -0700 Subject: [PATCH] Update compute_dp_sgd_privacy.py --- privacy/analysis/compute_dp_sgd_privacy.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/privacy/analysis/compute_dp_sgd_privacy.py b/privacy/analysis/compute_dp_sgd_privacy.py index 3357994..f0877c2 100644 --- a/privacy/analysis/compute_dp_sgd_privacy.py +++ b/privacy/analysis/compute_dp_sgd_privacy.py @@ -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))