From 72dba03ac31882d7678ac9e451ad027af81e23a4 Mon Sep 17 00:00:00 2001 From: lvmingfu Date: Mon, 26 Sep 2022 17:13:45 +0800 Subject: [PATCH] modify code formats for master --- mindarmour/privacy/sup_privacy/mask_monitor/masker.py | 10 +++++----- mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py | 10 +++++----- 2 files changed, 10 insertions(+), 10 deletions(-) diff --git a/mindarmour/privacy/sup_privacy/mask_monitor/masker.py b/mindarmour/privacy/sup_privacy/mask_monitor/masker.py index 85f1f4a..78ca5fe 100644 --- a/mindarmour/privacy/sup_privacy/mask_monitor/masker.py +++ b/mindarmour/privacy/sup_privacy/mask_monitor/masker.py @@ -36,8 +36,8 @@ class SuppressMasker(Callback): Examples: >>> import mindspore.nn as nn - >>> import mindspore.ops.operations as P - >>> from mindspore import context + >>> import mindspore as ms + >>> from mindspore import set_context, ops >>> from mindspore.nn import Accuracy >>> from mindarmour.privacy.sup_privacy import SuppressModel >>> from mindarmour.privacy.sup_privacy import SuppressMasker @@ -46,14 +46,14 @@ class SuppressMasker(Callback): >>> class Net(nn.Cell): ... def __init__(self): ... super(Net, self).__init__() - ... self._softmax = P.Softmax() + ... self._softmax = ops.Softmax() ... self._Dense = nn.Dense(10,10) - ... self._squeeze = P.Squeeze(1) + ... self._squeeze = ops.Squeeze(1) ... def construct(self, inputs): ... out = self._softmax(inputs) ... out = self._Dense(out) ... return self._squeeze(out) - >>> context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU") + >>> set_context(mode=ms.PYNATIVE_MODE, device_target="GPU") >>> network = Net() >>> masklayers = [] >>> masklayers.append(MaskLayerDes("_Dense.weight", 0, False, True, 10)) diff --git a/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py b/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py index 373fc87..6f82c2c 100644 --- a/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py +++ b/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py @@ -66,8 +66,8 @@ class SuppressPrivacyFactory: Examples: >>> import mindspore.nn as nn - >>> import mindspore.ops.operations as P - >>> from mindspore import context + >>> import mindspore as ms + >>> from mindspore import set_context, ops >>> from mindspore.nn import Accuracy >>> from mindarmour.privacy.sup_privacy import SuppressPrivacyFactory >>> from mindarmour.privacy.sup_privacy import MaskLayerDes @@ -75,14 +75,14 @@ class SuppressPrivacyFactory: >>> class Net(nn.Cell): ... def __init__(self): ... super(Net, self).__init__() - ... self._softmax = P.Softmax() + ... self._softmax = ops.Softmax() ... self._Dense = nn.Dense(10,10) - ... self._squeeze = P.Squeeze(1) + ... self._squeeze = ops.Squeeze(1) ... def construct(self, inputs): ... out = self._softmax(inputs) ... out = self._Dense(out) ... return self._squeeze(out) - >>> context.set_context(mode=context.PYNATIVE_MODE, device_target="CPU") + >>> set_context(mode=ms.PYNATIVE_MODE, device_target="CPU") >>> network = Net() >>> masklayers = [] >>> masklayers.append(MaskLayerDes("_Dense.weight", 0, False, True, 10))