3x3 convolution with padding
(in_planes: int, out_planes: int, stride: int = 1, groups: int = 1, dilation: int = 1)
| 24 | |
| 25 | |
| 26 | def conv3x3(in_planes: int, out_planes: int, stride: int = 1, groups: int = 1, dilation: int = 1) -> nn.Conv2d: |
| 27 | """3x3 convolution with padding""" |
| 28 | return nn.Conv2d(in_planes, out_planes, kernel_size=3, stride=stride, |
| 29 | padding=dilation, groups=groups, bias=False, dilation=dilation) |
| 30 | |
| 31 | |
| 32 | def conv1x1(in_planes: int, out_planes: int, stride: int = 1) -> nn.Conv2d: |