Python,torch.hub.load,模型加载失败问题

Python版本3.9,Torch版本1.13.1

问题描述:跑一个torch项目,代码中执行到加载模型时特别慢,然后报出了网络超时

具体代码如下

self.model = torch.hub.load('ultralytics/yolov5', 'custom', path=fileName)

经过查看发现,torch.hub.load是通过网上在线下载模型后加载,

因为众所周知的网络,该模型无法被在线下载,所以下载文件后,改为本地加载就可以正常运行了

self.model = torch.hub.load('./hub/ultralytics_yolov5_master', 'custom', path=fileName, source='local')

注意:torch.hub.load,该语句的模型默认下载地址:

C:\Users\你的用户名\.cache\torch\hub\ultralytics_yolov5_master,如果能在线下载的话,可以考虑从这里复制出来,我是在线下载后,将hub,复制到了项目同路径,所以可以用相对路径调用

  • 6
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值