Update rec_resnet_vd.py (#2681)

Fixed out_channels
2026-09-24 23:33:08 +08:00 · 2021-04-29 12:34:23 +08:00
parent 3b19311dc1
commit b40cefa6d2
1 changed files with 1 additions and 1 deletions
@@ -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