mxnet.gluon.model_zoo.model_store

Model zoo for pre-trained models.

Functions

get_model_file(name[, root])

Return location for the pretrained on local file system.

purge([root])

Purge all pretrained model files in local file store.

short_hash(name)

mxnet.gluon.model_zoo.model_store.get_model_file(name, root=None)[source]

Return location for the pretrained on local file system.

This function will download from online model zoo when model cannot be found or has mismatch. The root directory will be created if it doesn’t exist.

Parameters:
  • name (str) – Name of the model.

  • root (str, default $MXNET_HOME/models) – Location for keeping the model parameters.

Returns:

Path to the requested pretrained model file.

Return type:

file_path

mxnet.gluon.model_zoo.model_store.purge(root=None)[source]

Purge all pretrained model files in local file store.

Parameters:

root (str, default '$MXNET_HOME/models') – Location for keeping the model parameters.