Shortcuts

Source code for torchexpo.vision.image_classification.alexnet

import torchvision
from torchexpo.modules import ImageClassificationModule


[docs]def alexnet(): """AlexNet Model pre-trained on ImageNet""" model = torchvision.models.alexnet(pretrained=True) obj = ImageClassificationModule(model=model, model_name="AlexNet", 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.