| 201 | expansion = 1 |
| 202 | |
| 203 | def __init__(self, inplanes, planes, stride=1, downsample=None, groups=1, |
| 204 | base_width=64, dilation=1, norm_layer=None, |
| 205 | *, reduction=16): |
| 206 | super(SEBasicBlock, self).__init__() |
| 207 | if norm_layer is None: |
| 208 | norm_layer = nn.BatchNorm2d |
| 209 | self.conv1 = conv3x3(inplanes, planes, stride) |
| 210 | self.bn1 = norm_layer(planes) |
| 211 | self.relu = nn.ReLU(inplace=True) |
| 212 | self.conv2 = conv3x3(planes, planes, 1) |
| 213 | self.bn2 = norm_layer(planes) |
| 214 | self.se = SELayer(planes, reduction) |
| 215 | self.downsample = downsample |
| 216 | self.stride = stride |
| 217 | |
| 218 | def forward(self, x): |
| 219 | residual = x |