From b40cefa6d2b574b03bca7025f36251cb5126e772 Mon Sep 17 00:00:00 2001 From: YukSing <39983134+YukSing12@users.noreply.github.com> Date: Thu, 29 Apr 2021 12:34:23 +0800 Subject: [PATCH] Update rec_resnet_vd.py (#2681) Fixed out_channels --- ppocr/modeling/backbones/rec_resnet_vd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppocr/modeling/backbones/rec_resnet_vd.py b/ppocr/modeling/backbones/rec_resnet_vd.py index 6837ea0f..0187deb9 100644 --- a/ppocr/modeling/backbones/rec_resnet_vd.py +++ b/ppocr/modeling/backbones/rec_resnet_vd.py @@ -249,7 +249,7 @@ class ResNet(nn.Layer): name=conv_name)) shortcut = True self.block_list.append(bottleneck_block) - self.out_channels = num_filters[block] + self.out_channels = num_filters[block] * 4 else: for block in range(len(depth)): shortcut = False -- GitLab