[bug解决] torch.hub.load()无法联网,导致无法从github上在线加载模型

报错信息:

RuntimeError: It looks like there is no internet connection and the repo could not be found in the cache

报错位置:

 self.dinov2_vitg14 = torch.hub.load("facebookresearch/dinov2", "dinov2_vitg14", pretrained=False)

报错原因:

挂载服务器时,无法连接github,从而无法在线加载模型。

解决方案:

首先根据项目名称("facebookresearch/dinov2")将项目下载到本地,并传至服务器上。

然后将"facebookresearch/dinov2"修改为你的本地路径"path/to/your/dinov2_main"。

最后添加一个参数“source”

 self.dinov2_vitg14 = torch.hub.load("path/to/your/dinov2_main", "dinov2_vitg14", source='local' pretrained=False)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值