Browse Source

!382 fix an issue of diff-privacy

Merge pull request !382 from jxlang910/master
pull/349/MERGE
i-robot Gitee 2 years ago
parent
commit
46b4e56ec8
No known key found for this signature in database GPG Key ID: 173E9B9CA92EEF8F
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