Table Of Contents
Table Of Contents, **kwargs)[source]

Returns a pre-defined model by name

  • name (str) – Name of the model.

  • pretrained (bool) – Whether to load the pretrained weights for model.

  • classes (int) – Number of classes for the output layer.

  • 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.


The model.

Return type