torchvision.models — Torchvision 0.11.0 documentation
pytorch.org › vision › stableAlexnet¶ torchvision.models. alexnet (pretrained: bool = False, progress: bool = True, ** kwargs: Any) → torchvision.models.alexnet.AlexNet [source] ¶ AlexNet model architecture from the “One weird trick…” paper. The required minimum input size of the model is 63x63. Parameters. pretrained – If True, returns a model pre-trained on ...