How to access latest torchvision.models (e.g. ViT)?

vision
1

I have seen in the official torchvision docs that recently vision transformers and the ConvNeXt model families have been added to the PyTorch model zoo. However, even after upgrading to latest torchvision version 0.11.3 (via pip) these new models are not available:

>>> 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'

Any ideas how I can access these latest model additions in PyTorch?

2

The linked docs show the main branch which is currently at 0.13.0a0+d785158 so you might need to update torchvision to the latest nightly release.

3

Thanks a lot, that did the trick.
Do you have any ideas when 0.13 will officially be released?

4

v0.12 will be released officially this week, and includes ViT and ConvNeXt!

5

Thank you very much!