@@ -17,14 +17,13 @@ import os | |||||
import numpy as np | import numpy as np | ||||
from mindspore import Tensor | from mindspore import Tensor | ||||
from mindspore import context | from mindspore import context | ||||
from mindspore import nn | |||||
from mindspore import nn, Model | |||||
from mindspore.nn import SoftmaxCrossEntropyWithLogits | from mindspore.nn import SoftmaxCrossEntropyWithLogits | ||||
from mindspore.train import Model | |||||
from mindspore.train.callback import LossMonitor | from mindspore.train.callback import LossMonitor | ||||
from mindarmour.adv_robustness.attacks import FastGradientSignMethod | from mindarmour.adv_robustness.attacks import FastGradientSignMethod | ||||
from mindarmour.adv_robustness.defenses import NaturalAdversarialDefense | from mindarmour.adv_robustness.defenses import NaturalAdversarialDefense | ||||
from mindarmour.utils.logger import LogUtil | |||||
from mindarmour.utils import LogUtil | |||||
from examples.common.networks.lenet5.lenet5_net import LeNet5 | from examples.common.networks.lenet5.lenet5_net import LeNet5 | ||||
from examples.common.dataset.data_processing import generate_mnist_dataset | from examples.common.dataset.data_processing import generate_mnist_dataset | ||||
@@ -16,24 +16,23 @@ Training example of adaClip-mechanism differential privacy. | |||||
""" | """ | ||||
import os | import os | ||||
import mindspore.nn as nn | |||||
from mindspore import nn | |||||
from mindspore import context | from mindspore import context | ||||
from mindspore import load_checkpoint, load_param_into_net | |||||
from mindspore import dtype as mstype | |||||
from mindspore.train.callback import ModelCheckpoint | from mindspore.train.callback import ModelCheckpoint | ||||
from mindspore.train.callback import CheckpointConfig | from mindspore.train.callback import CheckpointConfig | ||||
from mindspore.train.callback import LossMonitor | from mindspore.train.callback import LossMonitor | ||||
from mindspore.nn.metrics import Accuracy | |||||
from mindspore.train.serialization import load_checkpoint, load_param_into_net | |||||
from mindspore.nn import Accuracy | |||||
import mindspore.dataset as ds | import mindspore.dataset as ds | ||||
import mindspore.dataset.vision.c_transforms as CV | import mindspore.dataset.vision.c_transforms as CV | ||||
import mindspore.dataset.transforms.c_transforms as C | import mindspore.dataset.transforms.c_transforms as C | ||||
from mindspore.dataset.vision import Inter | from mindspore.dataset.vision import Inter | ||||
import mindspore.common.dtype as mstype | |||||
from mindarmour.privacy.diff_privacy import DPModel | from mindarmour.privacy.diff_privacy import DPModel | ||||
from mindarmour.privacy.diff_privacy import PrivacyMonitorFactory | from mindarmour.privacy.diff_privacy import PrivacyMonitorFactory | ||||
from mindarmour.privacy.diff_privacy import NoiseMechanismsFactory | from mindarmour.privacy.diff_privacy import NoiseMechanismsFactory | ||||
from mindarmour.privacy.diff_privacy import ClipMechanismsFactory | from mindarmour.privacy.diff_privacy import ClipMechanismsFactory | ||||
from mindarmour.utils.logger import LogUtil | |||||
from mindarmour.utils import LogUtil | |||||
from examples.common.networks.lenet5.lenet5_net import LeNet5 | from examples.common.networks.lenet5.lenet5_net import LeNet5 | ||||
from lenet5_config import mnist_cfg as cfg | from lenet5_config import mnist_cfg as cfg | ||||
@@ -15,27 +15,28 @@ | |||||
Training example of suppress-based privacy. | Training example of suppress-based privacy. | ||||
""" | """ | ||||
import os | import os | ||||
import mindspore.nn as nn | import mindspore.nn as nn | ||||
from mindspore import context | from mindspore import context | ||||
from mindspore.train.callback import ModelCheckpoint | from mindspore.train.callback import ModelCheckpoint | ||||
from mindspore.train.callback import CheckpointConfig | from mindspore.train.callback import CheckpointConfig | ||||
from mindspore.train.callback import LossMonitor | from mindspore.train.callback import LossMonitor | ||||
from mindspore.nn.metrics import Accuracy | |||||
from mindspore.nn import Accuracy | |||||
import mindspore.dataset as ds | import mindspore.dataset as ds | ||||
import mindspore.dataset.vision.c_transforms as CV | import mindspore.dataset.vision.c_transforms as CV | ||||
import mindspore.dataset.transforms.c_transforms as C | import mindspore.dataset.transforms.c_transforms as C | ||||
from mindspore.dataset.vision.utils import Inter | |||||
import mindspore.common.dtype as mstype | |||||
from examples.common.networks.lenet5.lenet5_net import LeNet5 | |||||
from mindspore.dataset.vision import Inter | |||||
from mindspore import dtype as mstype | |||||
from sup_privacy_config import mnist_cfg as cfg | |||||
from mindarmour.privacy.sup_privacy import SuppressModel | from mindarmour.privacy.sup_privacy import SuppressModel | ||||
from mindarmour.privacy.sup_privacy import SuppressMasker | from mindarmour.privacy.sup_privacy import SuppressMasker | ||||
from mindarmour.privacy.sup_privacy import SuppressPrivacyFactory | from mindarmour.privacy.sup_privacy import SuppressPrivacyFactory | ||||
from mindarmour.privacy.sup_privacy import MaskLayerDes | from mindarmour.privacy.sup_privacy import MaskLayerDes | ||||
from mindarmour.utils import LogUtil | |||||
from examples.common.networks.lenet5.lenet5_net import LeNet5 | |||||
from sup_privacy_config import mnist_cfg as cfg | |||||
from mindarmour.utils.logger import LogUtil | |||||
LOGGER = LogUtil.get_instance() | LOGGER = LogUtil.get_instance() | ||||
LOGGER.set_level('INFO') | LOGGER.set_level('INFO') | ||||