Update rec_resnet_vd.py (PaddlePaddle#2681)

Fixed out_channels
Qs-Tim · Apr 29, 2021 · b40cefa · b40cefa
1 parent 3b19311
commit b40cefa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppocr/modeling/backbones/rec_resnet_vd.py b/ppocr/modeling/backbones/rec_resnet_vd.py
@@ -249,7 +249,7 @@ def __init__(self, in_channels=3, layers=50, **kwargs):
                             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