Browse Source

!175 Fix some issues for Suppress Privacy - modified on 2021.2.22

From: @itcomee
Reviewed-by: @jxlang910,@zhidanliu
Signed-off-by: @jxlang910
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
4ddcafd828
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