r"""ResNet-50 model from `"Deep Residual Learning for Image Recognition" `_ Args: pretrained (bool): If True, returns a model pre-trained on ImageNet progress (bool): If True, displays a progress bar of the download to stderr
(pretrained=False, progress=True, **kwargs)
| 255 | |
| 256 | |
| 257 | def resnet50(pretrained=False, progress=True, **kwargs): |
| 258 | r"""ResNet-50 model from |
| 259 | `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_ |
| 260 | Args: |
| 261 | pretrained (bool): If True, returns a model pre-trained on ImageNet |
| 262 | progress (bool): If True, displays a progress bar of the download to stderr |
| 263 | """ |
| 264 | return _resnet('resnet50', Bottleneck, [3, 4, 6, 3], pretrained, progress, |
| 265 | **kwargs) |
| 266 | |
| 267 | |
| 268 | def resnet101(pretrained=False, progress=True, **kwargs): |