未验证 提交 bd5d8509 编写于 作者: W wangguanzhong 提交者: GitHub

fix res5 (#2631)

上级 133375eb
...@@ -590,8 +590,9 @@ class Res5Head(nn.Layer): ...@@ -590,8 +590,9 @@ class Res5Head(nn.Layer):
if depth < 50: if depth < 50:
feat_in = 256 feat_in = 256
na = NameAdapter(self) na = NameAdapter(self)
block = BottleNeck if depth >= 50 else BasicBlock
self.res5 = Blocks( self.res5 = Blocks(
depth, feat_in, feat_out, count=3, name_adapter=na, stage_num=5) block, feat_in, feat_out, count=3, name_adapter=na, stage_num=5)
self.feat_out = feat_out if depth < 50 else feat_out * 4 self.feat_out = feat_out if depth < 50 else feat_out * 4
@property @property
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册