Unverified 提交 94a7f55c authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

FReLU bias=False bug fix (#1666)

上级 b2bef8f6
......@@ -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):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论