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)
| 86 | |
| 87 | |
| 88 | def resnet50(pretrained=False, progress=True, **kwargs): |
| 89 | r"""ResNet-50 model from |
| 90 | `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_ |
| 91 | |
| 92 | Args: |
| 93 | pretrained (bool): If True, returns a model pre-trained on ImageNet |
| 94 | progress (bool): If True, displays a progress bar of the download to stderr |
| 95 | """ |
| 96 | return _resnet('resnet50', Bottleneck, [3, 4, 6, 3], pretrained, progress, |
| 97 | **kwargs) |
| 98 | |
| 99 | |
| 100 | def resnet101(pretrained=False, progress=True, **kwargs): |