mindspore.hub

mindspore.hub.load_weights(network, network_name=None, force_reload=True, **kwargs)[source]

Load a model from mindspore, with pretrained weights.

Parameters

network (Cell) – Cell network.
network_name (string, optional) – Cell network name get from network. Default: None.
force_reload (bool, optional) – Whether to force a fresh download unconditionally. Default: False.
**kwargs (optional) – The corresponding kwargs for download for model. device_target (string, optional): Runtime device target. Default: ‘ascend’. dataset (string, optional): Dataset to train the network. Default: ‘cifar10’.

Example

>>> hub.load_weights(network, network_name='lenet',
                     **{'device_target': 'ascend', 'dataset':'mnist', 'version': '0.5.0'})