(self, dim, stride=1)
| 233 | nn.init.constant_(m.bias, 0) |
| 234 | |
| 235 | def _make_layer(self, dim, stride=1): |
| 236 | layer1 = BottleneckBlock(self.in_planes, dim, self.norm_fn, stride=stride) |
| 237 | layer2 = BottleneckBlock(dim, dim, self.norm_fn, stride=1) |
| 238 | layers = (layer1, layer2) |
| 239 | |
| 240 | self.in_planes = dim |
| 241 | return nn.Sequential(*layers) |
| 242 | |
| 243 | |
| 244 | def forward(self, x): |
no test coverage detected