(self, inplanes, planes, stride=1, downsample=None)
| 7 | |
| 8 | class BasicBlock(nn.Module): |
| 9 | def __init__(self, inplanes, planes, stride=1, downsample=None): |
| 10 | super(BasicBlock, self).__init__() |
| 11 | self.conv1 = conv1x1(inplanes, planes) |
| 12 | self.gn1 = GroupNorm(planes) |
| 13 | self.relu = nn.LeakyReLU(0.2, inplace=True) |
| 14 | self.conv2 = conv3x3(planes, planes, stride) |
| 15 | self.gn2 = GroupNorm(planes) |
| 16 | self.downsample = downsample |
| 17 | self.stride = stride |
| 18 | |
| 19 | def forward(self, x): |
| 20 | residual = x |