diff --git a/ppocr/modeling/necks/db_fpn.py b/ppocr/modeling/necks/db_fpn.py index 35adaff7c9361ab38d365f0eab88834effebfec4..f930bb5c45631ef6428a71d9a6d7d8fc1bdf6cf6 100644 --- a/ppocr/modeling/necks/db_fpn.py +++ b/ppocr/modeling/necks/db_fpn.py @@ -120,6 +120,7 @@ class CALayer(nn.Layer): def __init__(self, in_channels, out_channels, kernel_size, shortcut=True): super(CALayer, self).__init__() weight_attr = paddle.nn.initializer.KaimingUniform() + self.out_channels = out_channels self.in_conv = nn.Conv2D( in_channels=in_channels, out_channels=self.out_channels,