diff --git a/mindspore/hub.py b/mindspore/hub.py index f60e841729e97f3ce337fdfd5a3b85d4b7241976..52a1e7754f6537259bf2c2bc881fb61f142b2802 100644 --- a/mindspore/hub.py +++ b/mindspore/hub.py @@ -165,9 +165,11 @@ def load_weights(network, network_name=None, force_reload=True, **kwargs): 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'. + kwargs (dict, optional): The corresponding kwargs for download for model. + + - device_target (str, optional): Runtime device target. Default: 'ascend'. + - dataset (str, optional): Dataset to train the network. Default: 'cifar10'. + - version (str, optional): MindSpore version to save the checkpoint. Default: Latest version. Example: >>> hub.load(network, network_name='lenet',