Torch Hub download fails due to Torchvision dependency?

[jmwolf82]

🐛 Describe the bug

Running the following code returns the error seen below. It seems like there is some type of dependency or call to torchvision. Torchvision, however, doesn't list the new efficientnet models.

import torch
from torch import hub
#import torchvision.models as models

resnet18_model = torch.hub.load('pytorch/vision:main',
                         'resnet18',
                         pretrained = True)

It seems like there is something missing. I have cleared my .cache folder, and even updated the efficientnet.py code on my local machine to match the latest. Still receiving this Import Error. I am running the following versions:

Even if I do a simple:

torch.hub.list('pytorch/vision')

I receive the following error:

Versions

[pmeier]

I cannot reproduce. The only difference in my env is that I'm not on Windows.

My guess to what is happening: torch.hub commands add the downloaded repository to the Python path and afterwards import the package. Somehow this goes wrong and actually the version installed in your environment is loaded. The error then happens, because the hubconf.py from main contains entries

vision/hubconf.py

Lines 15 to 19 in 467b841

	efficientnet_b7,
	efficientnet_v2_s,
	efficientnet_v2_m,
	efficientnet_v2_l,
	)

that are not present in 0.12.0

vision/hubconf.py

Lines 15 to 16 in 9b5a3fe

	efficientnet_b7,
	)

Let's wait for @NicolasHug who knows torch.hub better than I do.

[jmwolf82]

@pmeier what OS are you using?

[pmeier]

Linux

[NicolasHug]

@jmwolf82 Could you try loading from 0.12.0 instead?

resnet18_model = torch.hub.load('pytorch/vision:v0.12.0',
                         'resnet18',
                         pretrained = True)

Note that since torchvision is already installed on your machine, you don't really need to use torchhub, you could just import resnet18 from torchvision.models directly.

[jmwolf82]

@NicolasHug that worked when I specifically referenced the branch in the command, I was able to use torch hub. It is interesting that I do not have the same issue on my Linux VM. Thanks for your help.

[pmeier]

It is interesting that I do not have the same issue on my Linux VM.

So torch.hub is indeed broken on Windows? We should look into that.

[NicolasHug]

I would suspect this is due to a different environment rather than a platform issue. But yeah I'm getting some other torchhub errors myself, something bad is going on. I can't do much about it right now because the torchhub repo's CI is failing for random reasons.