Skip to content

Y
yolov5
提交 c8892672 authored 作者: Glenn Jocher's avatar Glenn Jocher
浏览文件
操作

Funnel ReLU (FReLU) @staticmethod bug fix (#556)

Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
上级 131782a2
隐藏空白字符变更
内嵌 并排
正在显示 包含 0 行增加 和 1 行删除
utils/activations.py
浏览文件 @ c8892672
...@@ -69,6 +69,5 @@ class FReLU(nn.Module): ...@@ -69,6 +69,5 @@ class FReLU(nn.Module):
self.conv = nn.Conv2d(c1, c1, k, 1, 1, groups=c1) self.conv = nn.Conv2d(c1, c1, k, 1, 1, groups=c1)
self.bn = nn.BatchNorm2d(c1) self.bn = nn.BatchNorm2d(c1)
@staticmethod
def forward(self, x): def forward(self, x):
return torch.max(x, self.bn(self.conv(x))) return torch.max(x, self.bn(self.conv(x)))
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论