(self, block, inplanes, num_classes=1000,
groups=1, width_per_group=64, norm_layer=None)
| 42 | |
| 43 | class ResNetBase(nn.Module): |
| 44 | def __init__(self, block, inplanes, num_classes=1000, |
| 45 | groups=1, width_per_group=64, norm_layer=None): |
| 46 | super(ResNetBase, self).__init__() |
| 47 | |
| 48 | self._lock = threading.Lock() |
| 49 | self._block = block |
| 50 | self._norm_layer = nn.BatchNorm2d |
| 51 | self.inplanes = inplanes |
| 52 | self.dilation = 1 |
| 53 | self.groups = groups |
| 54 | self.base_width = width_per_group |
| 55 | |
| 56 | def _make_layer(self, planes, blocks, stride=1): |
| 57 | norm_layer = self._norm_layer |