Как получить доступ к последним моделям torchvision.models (например, ViT) ⇐ Python

[Anonymous]

В официальной документации torchvision я видел, что недавно в зоопарк моделей PyTorch были добавлены преобразователи зрения и семейства моделей ConvNeXt. Однако даже после обновления 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 "", line 1, in 
AttributeError: module 'torchvision.models' has no attribute 'vit_b_16'

Есть идеи, как получить доступ к этим последним дополнениям к моделям в PyTorch?


Подробнее здесь: https://stackoverflow.com/questions/713 ... ls-e-g-vit