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')