undo more whitespace changes

pull/217/head
Richard Brown 2024-05-06 16:28:24 +02:00
parent f95e1236bc
commit 089a830eea
1 changed files with 2 additions and 2 deletions

View File

@ -56,7 +56,7 @@ class Conv2dIBNormRelu(nn.Module):
class SEBlock(nn.Module):
""" SE Block Proposed in https://arxiv.org/pdf/1709.01507.pdf
""" SE Block Proposed in https://arxiv.org/pdf/1709.01507.pdf
"""
def __init__(self, in_channels, out_channels, reduction=1):
@ -68,7 +68,7 @@ class SEBlock(nn.Module):
nn.Linear(int(in_channels // reduction), out_channels, bias=False),
nn.Sigmoid()
)
def forward(self, x):
b, c, _, _ = x.size()
w = self.pool(x).view(b, c)