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