Browse Source

fix cmetrics problem

tags/v1.8.0
ZhidanLiu 3 years ago
parent
commit
a404c13fd4
3 changed files with 5 additions and 5 deletions
  1. +2
    -4
      mindarmour/privacy/diff_privacy/train/model.py
  2. +1
    -0
      mindarmour/privacy/sup_privacy/mask_monitor/masker.py
  3. +2
    -1
      mindarmour/privacy/sup_privacy/train/model.py

+ 2
- 4
mindarmour/privacy/diff_privacy/train/model.py View File

@@ -621,10 +621,8 @@ class _TrainOneStepCell(Cell):
self._noise_mech_param_updater = _MechanismsParamsUpdater(
decay_policy=self._noise_mech._decay_policy,
decay_rate=self._noise_mech._noise_decay_rate,
cur_noise_multiplier=
self._noise_mech._noise_multiplier,
init_noise_multiplier=
self._noise_mech._initial_noise_multiplier)
cur_noise_multiplier=self._noise_mech._noise_multiplier,
init_noise_multiplier=self._noise_mech._initial_noise_multiplier)

def construct(self, data, label):
"""


+ 1
- 0
mindarmour/privacy/sup_privacy/mask_monitor/masker.py View File

@@ -23,6 +23,7 @@ from mindarmour.privacy.sup_privacy.sup_ctrl.conctrl import SuppressCtrl
LOGGER = LogUtil.get_instance()
TAG = 'suppress masker'


class SuppressMasker(Callback):
"""
For details, please check `Tutorial <https://mindspore.cn/mindarmour/docs/zh-CN/master/protect_user_privacy_with_suppress_privacy.html#%E5%BC%95%E5%85%A5%E6%8A%91%E5%88%B6%E9%9A%90%E7%A7%81%E8%AE%AD%E7%BB%83>`_


+ 2
- 1
mindarmour/privacy/sup_privacy/train/model.py View File

@@ -184,6 +184,7 @@ class _TupleAdd(nn.Cell):
out = self.hyper_map(self.add, input1, input2)
return out


class _TupleMul(nn.Cell):
"""
Mul two tuple of data.
@@ -198,7 +199,7 @@ class _TupleMul(nn.Cell):
out = self.hyper_map(self.mul, input1, input2)
return out

# come from nn.cell_wrapper.TrainOneStepCell
class TrainOneStepCell(Cell):
r"""
Network training package class.


Loading…
Cancel
Save