diff --git a/utils/activations.py b/utils/activations.py index ba6b854dd..24f5a30f0 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):