In vit:如何访问最新的torchvision.models(例如ViT )

我在official torchvision docs中看到,最近视觉变压器和 ConvNeXt 模型系列已添加到 PyTorch 模型动物园。但是,即使升级到最新的 torchvision 版本 0.11.3(通过pip),这些新模型也不可用:

>>> import torchvision; torchvision.__version__
'0.11.3+cu102'
>>> import torchvision.models as models
>>> model = models.resnext50_32x4d()  # previous models work fine
>>> model = models.vit_b_16()  # vision transformers don't work
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: module 'torchvision.models' has no attribute 'vit_b_16'

任何想法我如何可以访问这些最新的模型添加在 PyTorch?

0

除了torchvision之外,还有其他获取 pytorch 模型的方法。您应该查看torch.hub,以获取通过此接口启用共享的特定 github repos 的模型。

此外,还有timm— — 许多 pytorch 视觉模型的存储库。

例如:

# list all ViT models
timm.list_models('vit_*')
# list all convNext models
timm.list_models('convnext*')
# load ViT-B/16
vit_b_16 = timm.create_model('vit_base_patch16_224', pretrained=True)
# load conv next
convnext = timm.create_model('convnext_base', pretrained=True)
0

事件虽然 @ Shai 的答案是一个很好的补充,我原来的问题是我如何可以访问官方的 ViT 和 ConvNeXt 模型在torchvision.models。事实证明,答案只是等待。所以对于记录:升级到版本 0.12 中的最新torchvisionpip 包后,我也得到了这些新模型。

本站系公益性非盈利分享网址,本文来自用户投稿,不代表边看边学立场,如若转载,请注明出处

(78)
Css那些事儿:BitCoin sentodaddress-那些该死的费用解释了吗
上一篇
C语言数据结构与算法:数据结构和算法书 C(what is data structure and algorithm)
下一篇

相关推荐

发表评论

登录 后才能评论

评论列表(16条)