我在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 包后,我也得到了这些新模型。
本站系公益性非盈利分享网址，本文来自用户投稿，不代表边看边学立场，如若转载，请注明出处
评论列表（68条）