From 0c36ef8260bd7fdd72d3f9124e6002a621c76ae9 Mon Sep 17 00:00:00 2001 From: jin-xiulang Date: Mon, 11 Jul 2022 10:46:30 +0800 Subject: [PATCH] fix an issue of diff-privacy --- mindarmour/privacy/diff_privacy/train/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindarmour/privacy/diff_privacy/train/model.py b/mindarmour/privacy/diff_privacy/train/model.py index c68c119..7af3480 100644 --- a/mindarmour/privacy/diff_privacy/train/model.py +++ b/mindarmour/privacy/diff_privacy/train/model.py @@ -137,7 +137,7 @@ class DPModel(Model): if clip_mech is None or isinstance(clip_mech, Cell): self._clip_mech = clip_mech - super(DPModel, self).__init__(**kwargs) + super(DPModel, self).__init__(optimizer=optimizer, **kwargs) # judge device_target, only GPU or Ascend is supported until now device = context.get_context("device_target")