diff --git a/mindspore/model_zoo/Bert_NEZHA/bert_for_pre_training.py b/mindspore/model_zoo/Bert_NEZHA/bert_for_pre_training.py index a5507fa782b8979cb02f996c410e22fca51c55cc..30a1c5290c5aa4e5074965b40ff2c5dc35226aaf 100644 --- a/mindspore/model_zoo/Bert_NEZHA/bert_for_pre_training.py +++ b/mindspore/model_zoo/Bert_NEZHA/bert_for_pre_training.py @@ -35,7 +35,6 @@ GRADIENT_CLIP_VALUE = 1.0 _nn_clip_by_norm = nn.ClipByNorm() clip_grad = C.MultitypeFuncGraph("clip_grad") @clip_grad.register("Number", "Number", "Tensor") - def _clip_grad(clip_type, clip_value, grad): """ Clip gradients.