From 883a5aff5a7f9bb9f22a5999088a769cb0e62431 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 10 Dec 2020 13:07:10 -0800 Subject: [PATCH] FReLU bias=False bug fix (#1607) --- utils/activations.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/activations.py b/utils/activations.py index ba6b854d..24f5a30f 100644 --- a/utils/activations.py +++ b/utils/activations.py @@ -65,7 +65,7 @@ class MemoryEfficientMish(nn.Module): class FReLU(nn.Module): def __init__(self, c1, k=3): # ch_in, kernel super().__init__() - self.conv = nn.Conv2d(c1, c1, k, 1, 1, groups=c1) + self.conv = nn.Conv2d(c1, c1, k, 1, 1, groups=c1, bias=False) self.bn = nn.BatchNorm2d(c1) def forward(self, x):