From 10c5773b9b5226b6406c72eafa4966dd407f1eba Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Tue, 26 Apr 2022 17:56:54 +0800 Subject: [PATCH] fix --- ppocr/modeling/necks/db_fpn.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ppocr/modeling/necks/db_fpn.py b/ppocr/modeling/necks/db_fpn.py index 35adaff7..f930bb5c 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, -- GitLab