3x3 convolution with padding
(in_planes: int, out_planes: int, stride: int = 1, groups: int = 1, dilation: int = 1)
| 8 | |
| 9 | |
| 10 | def conv3x3(in_planes: int, out_planes: int, stride: int = 1, groups: int = 1, dilation: int = 1) -> nn.Conv2d: |
| 11 | """3x3 convolution with padding""" |
| 12 | return nn.Conv2d(in_planes, out_planes, kernel_size=3, stride=stride, |
| 13 | padding=dilation, groups=groups, bias=False, dilation=dilation) |
| 14 | |
| 15 | |
| 16 | def conv1x1(in_planes: int, out_planes: int, stride: int = 1) -> nn.Conv2d: |