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: