From e64211ba9b8f13f583f72d4b8ce407729ec0f705 Mon Sep 17 00:00:00 2001 From: jin-xiulang Date: Wed, 24 Mar 2021 17:14:25 +0800 Subject: [PATCH] Remove the use of 'ControlDepend' in Diff privacy --- mindarmour/privacy/diff_privacy/train/model.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/mindarmour/privacy/diff_privacy/train/model.py b/mindarmour/privacy/diff_privacy/train/model.py index 5e6d53b..8bff84b 100644 --- a/mindarmour/privacy/diff_privacy/train/model.py +++ b/mindarmour/privacy/diff_privacy/train/model.py @@ -38,7 +38,6 @@ from mindspore.ops.operations import NPUAllocFloatStatus from mindspore.ops.operations import NPUClearFloatStatus from mindspore.ops.operations import ReduceSum from mindspore.ops.operations import LessEqual -from mindspore.ops.operations import ControlDepend from mindspore.parallel._utils import _get_gradients_mean from mindspore.parallel._utils import _get_device_num from mindspore.nn.wrap.grad_reducer import DistributedGradReducer @@ -395,7 +394,6 @@ class _TrainOneStepWithLossScaleCell(Cell): self.reduce_sum = ReduceSum(keep_dims=False) self.base = Tensor(1, mstype.float32) self.less_equal = LessEqual() - self.depend_parameter_use = ControlDepend(depend_mode=1) self.allreduce = P.AllReduce() self.parallel_mode = _get_parallel_mode() self.grad_reducer = F.identity