model_alexnet: AlexNet Model Architecture

In torchvision: Models, Datasets and Transformations for Images

AlexNet Model Architecture

Description

AlexNet model architecture from the One weird trick... paper.

Usage

model_alexnet(pretrained = FALSE, progress = TRUE, ...)

Arguments

pretrained	(bool): If TRUE, returns a model pre-trained on ImageNet.
progress	(bool): If TRUE, displays a progress bar of the download to stderr.
...	other parameters passed to the model intializer. currently only num_classes is used.

See Also

Other models: model_mobilenet_v2(), model_resnet

torchvision documentation built on June 22, 2024, 11:25 a.m.