1x1 convolution
(in_planes: int, out_planes: int, stride: int = 1)
| 14 | |
| 15 | |
| 16 | def conv1x1(in_planes: int, out_planes: int, stride: int = 1) -> nn.Conv2d: |
| 17 | """1x1 convolution""" |
| 18 | return nn.Conv2d(in_planes, out_planes, kernel_size=1, stride=stride, bias=False) |
| 19 | |
| 20 | |
| 21 | class BasicBlock(nn.Module): |
no outgoing calls
no test coverage detected