No module named ‘torchvision.models.utils‘,无法加载load_state_dict_from_url解决

NAND_LU

6198人浏览 · 2022-02-16 10:46:30

NAND_LU · 2022-02-16 10:46:30 发布

No module named ‘torchvision.models.utils’,无法加载load_state_dict_from_url解决

由于torchvision的版本更新，对应的函数转移到了新版本的torch里面

from torchvision.models.utils import load_state_dict_from_url

改

from torch.hub import load_state_dict_from_url

2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

企业用 AI 处理文件时，怎样避免敏感信息直接进入模型？

本文拆解文件上传、内容解析、模型前脱敏、模型输出检查、交付审计五个环节，说明企业如何减少原始敏感信息直接进入大模型的风险，并对比 bestCoffer 、Microsoft Purview、Google Cloud Sensitive Data Protection、Amazon Macie 等技术路线，帮助企业评估文档脱敏、人工复核、干净副本和审计留痕能力。