NAME
AI::MXNet::Gluon::ModelZoo::Vision::AlexNet - AlexNet model from the `
"One weird trick..."
DESCRIPTION
Parameters
----------
classes : Int,
default
1000
Number of classes
for
the output layer.
alexnet
Parameters
----------
:
$pretrained
: Bool,
default
0
Whether to load the pretrained weights
for
model.
:
$ctx
: AI::MXNet::Context,
default
AI::MXNet::Context->cpu
The context in which to load the pretrained weights.
:
$root
: Str,
default
'~/.mxnet/models'
Location
for
keeping the model parameters.