FReLU bias=False bug fix (#1666)
Browse files- utils/activations.py +1 -1
utils/activations.py
CHANGED
|
@@ -65,7 +65,7 @@ class MemoryEfficientMish(nn.Module):
|
|
| 65 |
class FReLU(nn.Module):
|
| 66 |
def __init__(self, c1, k=3): # ch_in, kernel
|
| 67 |
super().__init__()
|
| 68 |
-
self.conv = nn.Conv2d(c1, c1, k, 1, 1, groups=c1)
|
| 69 |
self.bn = nn.BatchNorm2d(c1)
|
| 70 |
|
| 71 |
def forward(self, x):
|
|
|
|
| 65 |
class FReLU(nn.Module):
|
| 66 |
def __init__(self, c1, k=3): # ch_in, kernel
|
| 67 |
super().__init__()
|
| 68 |
+
self.conv = nn.Conv2d(c1, c1, k, 1, 1, groups=c1, bias=False)
|
| 69 |
self.bn = nn.BatchNorm2d(c1)
|
| 70 |
|
| 71 |
def forward(self, x):
|