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.
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
more details about this class.
AlexNet model architecture from One weird trick for parallelizing convolutional neural networks.