Can't use convnext models

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

Any idea what might be going on? Thanks!

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

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