load¶
用于加载 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')