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

correct pafnet head (#3261)

上级 00848aea
...@@ -62,7 +62,6 @@ class HMHead(nn.Layer): ...@@ -62,7 +62,6 @@ class HMHead(nn.Layer):
in_channels=ch_in if i == 0 else ch_out, in_channels=ch_in if i == 0 else ch_out,
out_channels=ch_out, out_channels=ch_out,
norm_type=norm_type)) norm_type=norm_type))
head_conv.add_sublayer(lite_name + '.act', nn.ReLU6())
else: else:
if dcn_head: if dcn_head:
head_conv.add_sublayer( head_conv.add_sublayer(
...@@ -139,7 +138,6 @@ class WHHead(nn.Layer): ...@@ -139,7 +138,6 @@ class WHHead(nn.Layer):
in_channels=ch_in if i == 0 else ch_out, in_channels=ch_in if i == 0 else ch_out,
out_channels=ch_out, out_channels=ch_out,
norm_type=norm_type)) norm_type=norm_type))
head_conv.add_sublayer(lite_name + '.act', nn.ReLU6())
else: else:
if dcn_head: if dcn_head:
head_conv.add_sublayer( head_conv.add_sublayer(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册