Constructs a ResNet-50 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
(pretrained=True, **kwargs)
| 171 | |
| 172 | |
| 173 | def resnet50(pretrained=True, **kwargs): |
| 174 | """Constructs a ResNet-50 model. |
| 175 | Args: |
| 176 | pretrained (bool): If True, returns a model pre-trained on ImageNet |
| 177 | """ |
| 178 | model = ResNet(Bottleneck, [3, 4, 6, 3], **kwargs) |
| 179 | |
| 180 | return model |
| 181 | |
| 182 | |
| 183 | def resnet101(pretrained=True, **kwargs): |