get_model

get_model(name, **kwargs)

Returns a pre-defined model by name

Parameters:

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.

Returns:

gluon.HybridBlock

The model.