load

paddle.hub. load ( repo_dir, model, source='github', force_reload=False, **kwargs ) [源代码]

用于加载 repo 提供的功能/模型列表。

参数

  • repo_dir (str) - repo 地址,支持 git 地址形式和 local 地址。git 地址由 repo 拥有者/repo 名字:repo 分支组成,实例:PaddlePaddle/PaddleClas:develop;local 地址为 repo 的本地路径。

  • model (str)- 模型的名字。

  • source (str,可选) - 指定 repo 托管的位置,支持 github、gitee 和 local,默认值:github。

  • force_reload (bool,可选) - 指定是否强制拉取,默认值: False。

  • **kwargs (any,可选) - 模型参数。

返回

paddle.nn.Layer ,repo 提供的指定模型实例。

代码示例

>>> import paddle
>>> paddle.hub.load('lyuwenyu/paddlehub_demo:main', model='MM', source='github')