Skip to content

GitLab

Unverified Commit b40cefa6 authored by YukSing's avatar YukSing Committed by GitHub
Browse files

Update rec_resnet_vd.py (#2681) 

Fixed out_channels
parent 3b19311d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ppocr/modeling/backbones/rec_resnet_vd.py
View file @ b40cefa6
...@@ -249,7 +249,7 @@ class ResNet(nn.Layer): ...@@ -249,7 +249,7 @@ class ResNet(nn.Layer):
name=conv_name)) name=conv_name))
shortcut = True shortcut = True
self.block_list.append(bottleneck_block) self.block_list.append(bottleneck_block)
self.out_channels = num_filters[block] self.out_channels = num_filters[block] * 4
else: else:
for block in range(len(depth)): for block in range(len(depth)):
shortcut = False shortcut = False
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment