训练BIT模型时环境报错,ModuleNotFoundError: No module named ‘torchvision.models.utils’
这是由于torchvision版本过高。找到报错文件位置,找到报错代码行:
from torchvision.models.utils import load_state_dict_from_url
修改为:from torch.hub import load_state_dict_from_url
训练BIT模型时环境报错,ModuleNotFoundError: No module named ‘torchvision.models.utils’
这是由于torchvision版本过高。找到报错文件位置,找到报错代码行:
from torchvision.models.utils import load_state_dict_from_url
修改为:from torch.hub import load_state_dict_from_url