(
self, block: Type[nn.Module], planes: int, num_blocks: int, stride: int
)
| 64 | self.layer4 = self._make_layer(block, m_channels * 8, num_blocks[3], stride=2) |
| 65 | |
| 66 | def _make_layer( |
| 67 | self, block: Type[nn.Module], planes: int, num_blocks: int, stride: int |
| 68 | ) -> nn.Sequential: |
| 69 | strides = [stride] + [1] * (num_blocks - 1) |
| 70 | layers = [] |
| 71 | for stride in strides: |
| 72 | layers.append(block(self.in_planes, planes, stride)) |
| 73 | self.in_planes = planes * block.expansion |
| 74 | return nn.Sequential(*layers) |
| 75 | |
| 76 | def output_size(self) -> int: |
| 77 | return self._output_size |