FReLU bias=False bug fix (#1666)

2020-12-10 13:06:15 -08:00 · 2020-12-10 13:06:15 -08:00 · 94a7f55c4e
parent b2bef8f6d8
commit 94a7f55c4e
1 changed files with 1 additions and 1 deletions
--- 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):