mxnet.gluon.model_zoo.vision.alexnet¶

mxnet.gluon.model_zoo.vision.alexnet(pretrained=False, device=cpu(0), root=None, **kwargs)[source]¶

AlexNet model from the “One weird trick…” paper.

Parameters:

pretrained (bool, default False) – Whether to load the pretrained weights for model.
device (Device, default CPU) – The device in which to load the pretrained weights.
root (str, default $MXNET_HOME/models) – Location for keeping the model parameters.