From b7e44ebf94fd48d45f759e5a1aa51798328e04c7 Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Thu, 3 Jun 2021 14:33:59 +0800 Subject: [PATCH] correct pafnet head (#3261) --- ppdet/modeling/heads/ttf_head.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/ppdet/modeling/heads/ttf_head.py b/ppdet/modeling/heads/ttf_head.py index 02bb6d612..4d1360b34 100644 --- a/ppdet/modeling/heads/ttf_head.py +++ b/ppdet/modeling/heads/ttf_head.py @@ -62,7 +62,6 @@ class HMHead(nn.Layer): in_channels=ch_in if i == 0 else ch_out, out_channels=ch_out, norm_type=norm_type)) - head_conv.add_sublayer(lite_name + '.act', nn.ReLU6()) else: if dcn_head: head_conv.add_sublayer( @@ -139,7 +138,6 @@ class WHHead(nn.Layer): in_channels=ch_in if i == 0 else ch_out, out_channels=ch_out, norm_type=norm_type)) - head_conv.add_sublayer(lite_name + '.act', nn.ReLU6()) else: if dcn_head: head_conv.add_sublayer( -- GitLab