开发者社区

文档建议反馈控制台

最新优惠活动

文章/答案/技术大牛

发布

社区首页 >专栏 >PyTorch 1.0 中文文档：torch.hub

PyTorch 1.0 中文文档：torch.hub

ApacheCN_飞龙

发布于 2022-05-07 14:23:28

4260

发布于 2022-05-07 14:23:28

举报

文章被收录于专栏：信数据得永生信数据得永生

译者：kunwuz

torch.hub.load(github, model, force_reload=False, *args, **kwargs)

从github上加载一个带有预训练权重的模型。

参数:

github – 必需，一个字符串对象，格式为“repo_owner/repo_name[:tag_name]”，可选 tag/branch。如果未做指定，默认的 branch 是 master 。比方说: ‘pytorch/vision[:hub]’
model – 必须，一个字符串对象，名字在hubconf.py中定义。
force_reload – 可选，是否丢弃现有缓存并强制重新下载。默认是：False。
*args – 可选，可调用的model的相关args参数。
**kwargs – 可选，可调用的model的相关kwargs参数。

返回:	一个有相关预训练权重的单一模型。

torch.hub.set_dir(d)

也可以将hub_dir设置为本地目录来保存中间模型和检查点文件。

如果未设置此参数,环境变量TORCH_HUB_DIR 会被首先搜寻，~/.torch/hub 将被创建并用作后备。

阅读全文／改进本文

本文参与?腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2019-03-07，如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自作者个人站点/博客?前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与?腾讯云自媒体分享计划? ，欢迎热爱写作的你一起参与！

评论

登录后参与评论

0 条评论

热度

最新

LV.