Constructs a ResNet-101 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
(pretrained=False, **kwargs)
| 194 | |
| 195 | |
| 196 | def resnet101(pretrained=False, **kwargs): |
| 197 | """Constructs a ResNet-101 model. |
| 198 | |
| 199 | Args: |
| 200 | pretrained (bool): If True, returns a model pre-trained on ImageNet |
| 201 | """ |
| 202 | model = ResNet(Bottleneck, [3, 4, 23, 3], **kwargs) |
| 203 | if pretrained: |
| 204 | model.load_state_dict(load_url(model_urls['resnet101']), strict=False) |
| 205 | return model |
| 206 | |
| 207 | # def resnet152(pretrained=False, **kwargs): |
| 208 | # """Constructs a ResNet-152 model. |