Shortcuts

Source code for torchexpo.vision.image_classification.squeezenet

import torchvision
from torchexpo.modules import ImageClassificationModule


[docs]def squeezenet1_0(): """SqueezeNet 1.0 Model pre-trained on ImageNet""" model = torchvision.models.squeezenet1_0(pretrained=True) obj = ImageClassificationModule(model, "SqueezeNet1_0", model_example="default") return obj
[docs]def squeezenet1_1(): """SqueezeNet 1.1 Model pre-trained on ImageNet""" model = torchvision.models.squeezenet1_1(pretrained=True) obj = ImageClassificationModule(model, "SqueezeNet1_1", 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.