diff --git a/ppocr/modeling/heads/rec_ctc_head.py b/ppocr/modeling/heads/rec_ctc_head.py index 27c8c7c74fb5a6fe378f136aea7a792ac187a9ac..69d4ef50b648c0251b9b8d0b4c1e731a6f236105 100755 --- a/ppocr/modeling/heads/rec_ctc_head.py +++ b/ppocr/modeling/heads/rec_ctc_head.py @@ -24,7 +24,7 @@ from paddle.nn import functional as F def get_para_bias_attr(l2_decay, k, name): - regularizer = paddle.fluid.regularizer.L2Decay(l2_decay) + regularizer = paddle.regularizer.L2Decay(l2_decay) stdv = 1.0 / math.sqrt(k * 1.0) initializer = nn.initializer.Uniform(-stdv, stdv) weight_attr = ParamAttr( @@ -33,6 +33,7 @@ def get_para_bias_attr(l2_decay, k, name): regularizer=regularizer, initializer=initializer, name=name + "_b_attr") return [weight_attr, bias_attr] + class CTCHead(nn.Layer): def __init__(self, in_channels, out_channels, fc_decay=0.0004, **kwargs): super(CTCHead, self).__init__() diff --git a/ppocr/optimizer/regularizer.py b/ppocr/optimizer/regularizer.py index 8ac1b81f60677ebb5c4d9ef034fc6f84d97fd5f8..c6396f338d9d40fc444083e205fd55329e7dfd59 100644 --- a/ppocr/optimizer/regularizer.py +++ b/ppocr/optimizer/regularizer.py @@ -17,7 +17,7 @@ from __future__ import division from __future__ import print_function from __future__ import unicode_literals -from paddle import fluid +import paddle class L1Decay(object): @@ -32,8 +32,7 @@ class L1Decay(object): self.regularization_coeff = factor def __call__(self): - reg = fluid.regularizer.L1Decay( - regularization_coeff=self.regularization_coeff) + reg = paddle.regularizer.L1Decay(self.regularization_coeff) return reg @@ -49,6 +48,5 @@ class L2Decay(object): self.regularization_coeff = factor def __call__(self): - reg = fluid.regularizer.L2Decay( - regularization_coeff=self.regularization_coeff) + reg = paddle.regularizer.L2Decay(self.regularization_coeff) return reg