(self, dim, stride=1)
| 157 | nn.init.constant_(m.bias, 0) |
| 158 | |
| 159 | def _make_layer(self, dim, stride=1): |
| 160 | layer1 = ResidualBlock(self.in_planes, dim, self.norm_fn, stride=stride) |
| 161 | layer2 = ResidualBlock(dim, dim, self.norm_fn, stride=1) |
| 162 | layers = (layer1, layer2) |
| 163 | |
| 164 | self.in_planes = dim |
| 165 | return nn.Sequential(*layers) |
| 166 | |
| 167 | |
| 168 | def forward(self, x): |