diff --git a/mindarmour/privacy/sup_privacy/mask_monitor/masker.py b/mindarmour/privacy/sup_privacy/mask_monitor/masker.py index 1a54ae9..786097d 100644 --- a/mindarmour/privacy/sup_privacy/mask_monitor/masker.py +++ b/mindarmour/privacy/sup_privacy/mask_monitor/masker.py @@ -23,6 +23,7 @@ from mindarmour.privacy.sup_privacy.sup_ctrl.conctrl import SuppressCtrl LOGGER = LogUtil.get_instance() TAG = 'suppress masker' + class SuppressMasker(Callback): """ Periodicity check suppress privacy function status and toggle suppress operation. diff --git a/mindarmour/privacy/sup_privacy/train/model.py b/mindarmour/privacy/sup_privacy/train/model.py index ac49434..5b1a34f 100644 --- a/mindarmour/privacy/sup_privacy/train/model.py +++ b/mindarmour/privacy/sup_privacy/train/model.py @@ -186,6 +186,7 @@ class _TupleAdd(nn.Cell): out = self.hyper_map(self.add, input1, input2) return out + class _TupleMul(nn.Cell): """ Mul two tuple of data. @@ -200,6 +201,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"""