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

fix gfl export model (#7462)

上级 e309cf72
......@@ -139,7 +139,7 @@ class DGQP(nn.Layer):
axis=2, keepdim=True)], axis=2)
else:
stat = prob_topk
y = F.relu(self.reg_conv1(stat.reshape([N, -1, H, W])))
y = F.relu(self.reg_conv1(stat.reshape([N, 4 * self.reg_topk, H, W])))
y = F.sigmoid(self.reg_conv2(y))
return y
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册