search

mustafaemreyildirim / Age-Gender-Estimator

6 stars 0 forks source link

can not load pytorch/vision:0.5.0 #1

Open codermckee opened 8 months ago

codermckee commented 8 months ago

error in test.py line9: extractor = torch.hub.load('pytorch/vision:v0.5.0', 'mobilenet_v2', pretrained=True).features

mustafaemreyildirim commented 4 months ago

Probably, there is a conflict on version of Pytorch. try to add below: from torch import inf