facebookresearch/swav

Cannot load the pretrained model with torch.hub.load

TuanDTr opened this issue · 0 comments

Hi, I tried to load the pretrained model from model zoo with torch.hub and got the following error:

from src.resnet50 import resnet50w2 as _resnet50w2
ModuleNotFoundError: No module named 'src.resnet50'