From f80e57964239488f28ffc4be28fba838bb813fd7 Mon Sep 17 00:00:00 2001 From: chenzomi Date: Mon, 3 Aug 2020 19:57:43 +0800 Subject: [PATCH] change hub --- mindspore/hub.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/mindspore/hub.py b/mindspore/hub.py index f60e84172..52a1e7754 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', -- GitLab