search

MadryLab / robustness

A library for experimenting with, training and evaluating neural networks, with a focus on adversarial robustness.
MIT License
912 stars 181 forks source link

import robustness. dataset "No module named 'torchvision.models.utils'" #119

Open yifan6210 opened 1 year ago

yifan6210 commented 1 year ago

Hi, package developers,

I encountered an issue with the import robustness. dataset and I think the potential problem occurs in the vgg.py file under imagenet models.

potential changes from

from torchvision.models.utils import load_state_dict_from_url

to

from torch.hub import load_state_dict_from_url

allanzhou777 commented 1 year ago

This was updated in github but not in PyPi

pstefanou12 commented 1 year ago

To fix this, the owner of the repository just needs to push the updates to PyPi.