(self, channels, reduction=16)
| 153 | |
| 154 | class SqueezeExcitation(nn.Module): |
| 155 | def __init__(self, channels, reduction=16): |
| 156 | super().__init__() |
| 157 | self.fc1 = nn.Linear(channels, channels // reduction) |
| 158 | self.fc2 = nn.Linear(channels // reduction, channels) |
| 159 | |
| 160 | def forward(self, x): |
| 161 | b, c, _, _ = x.size() |