!337 fix cmetric problem

Merge pull request !337 from ZhidanLiu/master

mindarmour/privacy/diff_privacy/train/model.py

@@ -622,10 +622,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):
         """

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.

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,7 +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"""
     Network training package class.