Constructs a ResNet-50 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
(pretrained=False, **kwargs)
| 182 | ''' |
| 183 | |
| 184 | def resnet50(pretrained=False, **kwargs): |
| 185 | """Constructs a ResNet-50 model. |
| 186 | |
| 187 | Args: |
| 188 | pretrained (bool): If True, returns a model pre-trained on ImageNet |
| 189 | """ |
| 190 | model = ResNet(Bottleneck, [3, 4, 6, 3], **kwargs) |
| 191 | if pretrained: |
| 192 | model.load_state_dict(load_url(model_urls['resnet50']), strict=False) |
| 193 | return model |
| 194 | |
| 195 | |
| 196 | def resnet101(pretrained=False, **kwargs): |