megengine.hub.load#

load(repo_info, entry, *args, git_host=DEFAULT_GIT_HOST, use_cache=True, commit=None, protocol=DEFAULT_PROTOCOL, **kwargs)[源代码]#

从GitHub或GitLab中加载具有预训练权重的模型。

参数:

  • repo_info (str) – 格式为``”repo_owner/repo_name[:tag_name/:branch_name]”的字符串，其中可选参数为batch。如果没有特殊指定默认的分支为``master，例如："brain_sdk/MegBrain[:hub]"

  • entry (str) – 一个在hubconf中定义的入口点。

  • git_host (str) – git仓库的主地址。例如：github.com

  • use_cache (bool) – 选择使用本地缓存的代码或完全重新拉取代码。

  • commit (Optional[str]) – GitHub或GitLab的commit id。

  • protocol (str) – 获得代码仓库所使用的协议，其中，HTTPS协议只支持github公共仓库。该参数值可为HTTPS 或 SSH。

返回类型:

Any

返回:

单个模型，具有对应的预训练的权重。