Browse Source

fix an issue of diff-privacy

pull/382/head
jin-xiulang 2 years ago
parent
commit
0c36ef8260
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindarmour/privacy/diff_privacy/train/model.py

+ 1
- 1
mindarmour/privacy/diff_privacy/train/model.py View File

@@ -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")


Loading…
Cancel
Save