Shortcuts

AlexNet

The AlexNet model was originally introduced in the ImageNet Classification with Deep Convolutional Neural Networks paper. The implemented architecture is slightly different from the original one, and is based on One weird trick for parallelizing convolutional neural networks.

Model builders

The following model builders can be used to instanciate an AlexNet model, with or without pre-trained weights. All the model builders internally rely on the torchvision.models.alexnet.AlexNet base class. Please refer to the source code for more details about this class.

alexnet(*[, weights, progress])

AlexNet model architecture from One weird trick for parallelizing convolutional neural networks.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources