未验证 提交 b40cefa6 编写于 作者: Y YukSing 提交者: GitHub

Update rec_resnet_vd.py (#2681)

Fixed out_channels
上级 3b19311d
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册