From a4b1993c842f8b9c56ed6d7544da22d1ad606ba8 Mon Sep 17 00:00:00 2001 From: Chris Ha <15088501+VRandme@users.noreply.github.com> Date: Tue, 11 Feb 2020 19:35:06 +0900 Subject: [PATCH] Restore Bias The bias term might be useful for actual spatial 2d convs --- timm/models/EcaModule.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/EcaModule.py b/timm/models/EcaModule.py index 09ef0038..54971436 100644 --- a/timm/models/EcaModule.py +++ b/timm/models/EcaModule.py @@ -130,7 +130,7 @@ class SpatialGate(nn.Module): super(SpatialGate, self).__init__() kernel_size = 7 self.compress = ChannelPool() - self.spatial = nn.Conv2d(2, 1, kernel_size, stride=1, padding=(kernel_size-1) // 2, bias=False) + self.spatial = nn.Conv2d(2, 1, kernel_size, stride=1, padding=(kernel_size-1) // 2) def forward(self, x): x_compress = self.compress(x) x_out = self.spatial(x_compress)