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和local,默认值:github
force_reload (bool | 可选) : 指定是否强制拉取,默认值: False
kwargs (any | 可选) : 模型参数
返回
paddle.nn.Layer
,repo提供的指定模型实例
代码示例
import paddle
model = paddle.hub.help('PaddlePaddle/PaddleClas:develop', 'alexnet', source='github', force_reload=True)
print(model)