3x3 convolution with padding
(in_planes, out_planes, stride=1, groups=1, dilation=1)
| 15 | |
| 16 | |
| 17 | def conv3x3(in_planes, out_planes, stride=1, groups=1, dilation=1): |
| 18 | """3x3 convolution with padding""" |
| 19 | return nn.Conv2d( |
| 20 | in_planes, |
| 21 | out_planes, |
| 22 | kernel_size=3, |
| 23 | stride=stride, |
| 24 | padding=dilation, |
| 25 | groups=groups, |
| 26 | bias=False, |
| 27 | dilation=dilation, |
| 28 | ) |
| 29 | |
| 30 | |
| 31 | class BasicBlock(nn.Module): |