mxnet.gluon.model_zoo.vision.alexnet¶
-
mxnet.gluon.model_zoo.vision.
alexnet
(pretrained=False, ctx=cpu(0), root='/var/lib/jenkins/.mxnet/models', **kwargs)[source]¶ AlexNet model from the “One weird trick…” paper.
- Parameters
pretrained (bool, default False) – Whether to load the pretrained weights for model.
ctx (Context, default CPU) – The context in which to load the pretrained weights.
root (str, default $MXNET_HOME/models) – Location for keeping the model parameters.