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,复制到了项目同路径,所以可以用相对路径调用