apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

FReLU bias=False bug fix (#1607)

Browse Source
This commit is contained in:
Glenn Jocher 2020-12-10 13:07:10 -08:00 committed by GitHub
parent 61fb2dbd20
commit 883a5aff5a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/activations.py
View File

@ -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):