Shortcuts

Source code for torchexpo.vision.image_classification.wide_resnet

import torchvision
from torchexpo.modules import ImageClassificationModule


[docs]def wide_resnet50_2(): """Wide ResNet-50-2 Model pre-trained on ImageNet""" model = torchvision.models.wide_resnet50_2(pretrained=True) obj = ImageClassificationModule(model, "Wide ResNet50 2", model_example="default") return obj
[docs]def wide_resnet101_2(): """Wide ResNet-101-2 Model pre-trained on ImageNet""" model = torchvision.models.wide_resnet101_2(pretrained=True) obj = ImageClassificationModule(model, "Wide ResNet101 2", model_example="default") return obj
Read the Docs v: latest
Versions
latest
Downloads
pdf
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.