AlexNet Model Architecture — model_alexnet • torchvision

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

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_efficientnet, model_efficientnet_v2, model_inception_v3(), model_mobilenet_v2(), model_resnet, model_vgg, model_vit