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) – a string with format "repo_owner/repo_name[:tag_name/:branch_name]" with an optional tag/branch. The default branch is master if not specified. Eg: "brain_sdk/MegBrain[:hub]"
entry (str) – 一个在hubconf中定义的入口点。
git_host (str) – host address of git repo. Eg: github.com
use_cache (bool) – 选择使用本地缓存的代码或完全重新拉取代码。
commit (Optional[str]) – GitHub或GitLab的commit id。
protocol (str) – 获得代码仓库所使用的协议，其中，HTTPS协议只支持github公共仓库。该参数值可为HTTPS 或 SSH。

返回类型

Any

返回

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

megengine.hub.list

megengine.hub.help