Browse Source

suppress based privacy model, 2021.2.22

tags/v1.2.0-rc1
itcomee 4 years ago
parent
commit
56385aba03
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py

+ 1
- 0
mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py View File

@@ -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()


Loading…
Cancel
Save