wangtao
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

change hub

This commit is contained in:
chenzomi 2020-08-03 19:57:43 +08:00
parent 1091cdeedf
commit f80e579642
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/hub.py
View File

@ -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',