torchvision.models — Torchvision 0.8.1 documentation
pytorch.org › vision › 0torchvision.models.resnet101(pretrained=False, progress=True, **kwargs) [source] ResNet-101 model from “Deep Residual Learning for Image Recognition” Parameters: pretrained ( bool) – If True, returns a model pre-trained on ImageNet progress ( bool) – If True, displays a progress bar of the download to stderr
resnet50 — Torchvision main documentation
pytorch.org › generated › torchvisiontorchvision.models.resnet50(pretrained: bool = False, progress: bool = True, **kwargs: Any) → torchvision.models.resnet.ResNet [source] ResNet-50 model from “Deep Residual Learning for Image Recognition”. Parameters. pretrained ( bool) – If True, returns a model pre-trained on ImageNet. progress ( bool) – If True, displays a progress ...