Browse Source

fix an issue of diff-privacy

tags/v1.9.0
jin-xiulang 3 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): if clip_mech is None or isinstance(clip_mech, Cell):
self._clip_mech = clip_mech 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 # judge device_target, only GPU or Ascend is supported until now
device = context.get_context("device_target") device = context.get_context("device_target")


Loading…
Cancel
Save