PyTorch Hub 正在参加 2020 年度 OSC 中国开源项目评选,请投票支持!
PyTorch Hub 在 2020 年度 OSC 中国开源项目评选 中已获得 {{ projectVoteCount }} 票,请投票支持!
投票让它出道
已投票
授权协议: 未知
开发语言: Python
操作系统: 跨平台
开发厂商: Facebook
收录时间: 2019-06-11
提 交 者: 红薯

Facebook宣布推出PyTorch Hub,一个包含计算机视觉、自然语言处理领域的诸多经典模型的聚合中心,让你调用起来更方便。

PyTorch Hub是一个简易API和工作流程,为复现研究提供了基本构建模块,包含预训练模型库

并且,PyTorch Hub还支持Colab,能与论文代码结合网站Papers With Code集成,用于更广泛的研究。

发布首日已有18个模型“入驻”,获得英伟达官方力挺。而且Facebook还鼓励论文发布者把自己的模型发布到这里来,让PyTorch Hub越来越强大。

PyTorch Hub的使用简单到不能再简单,不需要下载模型,只用了一个torch.hub.load()就完成了对图像分类模型AlexNet的调用。

PyTorch Hub允许用户对已发布的模型执行以下操作:

1、查询可用的模型;
2、加载模型;
3、查询模型中可用的方法。

下面让我们来看看每个应用的实例。

1、查询可用的模型

用户可以使用torch.hub.list()这个API列出repo中所有可用的入口点。比如你想知道PyTorch Hub中有哪些可用的计算机视觉模型:

>>> torch.hub.list('pytorch/vision')
>>>
['alexnet',
'deeplabv3_resnet101',
'densenet121',
...
'vgg16',
'vgg16_bn',
'vgg19',
 'vgg19_bn']

2、加载模型

在上一步中能看到所有可用的计算机视觉模型,如果想调用其中的一个,也不必安装,只需一句话就能加载模型。

model = torch.hub.load('pytorch/vision', 'deeplabv3_resnet101', pretrained=True)

至于如何获得此模型的详细帮助信息,可以使用下面的API:

print(torch.hub.help('pytorch/vision', 'deeplabv3_resnet101'))

如果模型的发布者后续加入错误修复和性能改进,用户也可以非常简单地获取更新,确保自己用到的是最新版本:

model = torch.hub.load(..., force_reload=True)

对于另外一部分用户来说,稳定性更加重要,他们有时候需要调用特定分支的代码。例如pytorch_GAN_zoo的hub分支:

model = torch.hub.load('facebookresearch/pytorch_GAN_zoo:hub', 'DCGAN', pretrained=True, useGPU=False)

3、查看模型可用方法

从PyTorch Hub加载模型后,你可以用dir(model)查看模型的所有可用方法。以bertForMaskedLM模型为例:

>>> dir(model)
>>>
['forward'
...
'to'
'state_dict',
]

如果你对forward方法感兴趣,使用help(model.forward) 了解运行运行该方法所需的参数。

>>> help(model.forward)
>>>
Help on method forward in module pytorch_pretrained_bert.modeling:
forward(input_ids, token_type_ids=None, attention_mask=None, masked_lm_labels=None)
...

PyTorch Hub中提供的模型也支持Colab。

进入每个模型的介绍页面后,你不仅可以看到GitHub代码页的入口,甚至可以一键进入Colab运行模型Demo。

介绍内容来自量子位

展开阅读全文

代码

的 Gitee 指数为
超过 的项目

PyTorch Hub 的相关资讯

还没有任何资讯

PyTorch Hub 的相关博客

[pytorch] PyTorch Hook

PyTorch Hook¶ 为什么要引入hook? -> hook可以做什么? 都有哪些hook? 如何使用hook? 1. 为什么引入hook?¶ 参考:Pytorch中...

pytorch学习-WHAT IS PYTORCH

参考:https://pytorch.org/tutorials/beginner/blitz/tensor_tutorial.html#sphx-glr-beginner-blitz-tensor-tutorial-py WH...

[pytorch] Pytorch入门

Pytorch入门 简单容易上手,感觉比keras好理解多了,和mxnet很像(似乎mxnet有点借鉴pytorch),记一记。 直接从例子开始学,...

【Pytorch】Pytorch 中的 dim

Pytorch 中对 tensor 的很多操作如sum、argmax、等都可以设置dim参数用来指定操作在哪一维进行。Pytorch 中的 dim 类似于 nu...

【pytorch】pytorch基础学习

[TOC] 1. 前言   最近在学习pytorch,先照着官方的“60分钟教程”学习了一下,然后再github上找了两个star比较多的项目,自...

【PyTorch】PyTorch中的梯度累加

PyTorch中的梯度累加 使用PyTorch实现梯度累加变相扩大batch PyTorch中在反向传播前为什么要手动将梯度清零? - Pascal的回答...

PyTorch-22 学习 PyTorch 的 Examples

https://www.cnblogs.com/studyai/p/12488568.html 本教程通过自包含的例子介绍 PyTorch 的基本概念。 要查看格式更加优美的图...

PyTorch-22 学习 PyTorch 的 Examples

https://www.cnblogs.com/studyai/p/12488568.html 本教程通过自包含的例子介绍 PyTorch 的基本概念。 要查看格式更加优美的图...

【PyTorch深度学习】学习笔记之PyTorch与深度学习

第1章 PyTorch与深度学习 深度学习的应用 接近人类水平的图像分类 接近人类水平的语音识别 机器翻译 自动驾驶汽车 Siri、Goo...

Pytorch

一、pytorch张量数据类型: 1.python数据类型与pytorch数据类型的对应关系: python pytorch Int IntTensor of size() float...

PyTorch Hub 的相关问答

还没有任何问答,马上提问

评论 (0)

加载中
更多评论
0 评论
16 收藏
分享
返回顶部
顶部