BIT_CD
BIT_CD copied to clipboard
python demo.py ModuleNotFoundError: No module named 'torchvision.models.utils'
when i run demo.py file i have the error ModuleNotFoundError: No module named 'torchvision.models.utils'
Same here. Did you ever figure it out?
错误原因:torchvision的版本问题 解决方案:将BIT_CD-master\models\resnet.py第3行from torchvision.models.utils import load_state_dict_from_url 替换为from torch.hub import load_state_dict_from_url