Browse Source

!38 fix issue

Merge pull request !38 from zheng-huanhuan/master
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
bfd8d88d6d
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindarmour/diff_privacy/train/model.py

+ 2
- 2
mindarmour/diff_privacy/train/model.py View File

@@ -114,8 +114,8 @@ class DPModel(Model):
self._micro_batches = check_int_positive('micro_batches', micro_batches)
else:
self._micro_batches = None
float_norm_clip = check_param_type('l2_norm_clip', norm_clip, float)
self._norm_clip = check_value_positive('l2_norm_clip', float_norm_clip)
norm_clip = check_param_type('norm_clip', norm_clip, float)
self._norm_clip = check_value_positive('norm_clip', norm_clip)
if isinstance(dp_mech, mechanisms.Mechanisms):
self._dp_mech = dp_mech
else:


Loading…
Cancel
Save