From 56385aba03dc1902a67c55ebc8365d47c1f34ffe Mon Sep 17 00:00:00 2001 From: itcomee Date: Mon, 22 Feb 2021 19:27:29 +0800 Subject: [PATCH] suppress based privacy model, 2021.2.22 --- mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py b/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py index 8a45dd9..7244ee0 100644 --- a/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py +++ b/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py @@ -300,6 +300,7 @@ class SuppressCtrl(Cell): for layer in networks.get_parameters(expand=True): grad_idx = self.grad_idx_map[m] if grad_idx < 0: + m = m + 1 continue if self.grads_mask_list[grad_idx].mask_able: weight_array = layer.data.asnumpy()