Can't use convnext models

vision
1

Hi all,

I am trying to use ConvNeXt models in my implementation, but everytime I call the model (either it’s the tiny, base, or whatever) I get the following error:

self.model = models.convnext_tiny(pretrained=True)
AttributeError: module 'torchvision.models' has no attribute 'convnext_tiny'

The last torch installation I have was made using:

pip3 install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio===0.10.1+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

Any idea what might be going on? Thanks!

2

It seems this model just landed 28 days ago in this PR so you would need to install the nightly release, not the stable one.

1 Like
3

Great, I was figuring it had to do something with the version. Thanks a lot!