未验证 提交 8ffd18b2 编写于 作者: W Walter 提交者: GitHub

Merge pull request #748 from weisy11/develop_reg

modify hrnet
......@@ -37,7 +37,7 @@ def forward(self, inputs):
y = self.pool2d_max(y)
for bottleneck_block in self.bottleneck_block_list:
y = bottleneck_block(y)
y = self.pool2d_avg(y)
y = self.avg_pool(y)
y = fluid.layers.reshape(y, shape=[-1, self.pool2d_avg_output])
y = self.out(y)
return y, self.fm
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册