From 2003b91e01414f8405e001212650b8766912d204 Mon Sep 17 00:00:00 2001 From: zhenghuanhuan Date: Thu, 18 Jun 2020 19:30:30 +0800 Subject: [PATCH] [MA][diff_privacy][Func] AdaGaussianRandom not checked param alpha https://gitee.com/mindspore/dashboard?issue_id=I1J496 --- mindarmour/diff_privacy/train/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindarmour/diff_privacy/train/model.py b/mindarmour/diff_privacy/train/model.py index 17c4927..1ad1675 100644 --- a/mindarmour/diff_privacy/train/model.py +++ b/mindarmour/diff_privacy/train/model.py @@ -114,8 +114,8 @@ class DPModel(Model): self._micro_batches = check_int_positive('micro_batches', micro_batches) else: self._micro_batches = None - float_norm_clip = check_param_type('l2_norm_clip', norm_clip, float) - self._norm_clip = check_value_positive('l2_norm_clip', float_norm_clip) + norm_clip = check_param_type('norm_clip', norm_clip, float) + self._norm_clip = check_value_positive('norm_clip', norm_clip) if isinstance(dp_mech, mechanisms.Mechanisms): self._dp_mech = dp_mech else: -- GitLab