mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

FReLU bias=False bug fix (#1666)

pull/1677/head
Glenn Jocher 2020-12-10 13:06:15 -08:00 committed by GitHub
parent b2bef8f6d8
commit 94a7f55c4e
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): class FReLU(nn.Module):
def __init__(self, c1, k=3): # ch_in, kernel def __init__(self, c1, k=3): # ch_in, kernel
super().__init__() 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) self.bn = nn.BatchNorm2d(c1)
def forward(self, x): def forward(self, x):