小编cod*_*com的帖子

如何使用 torch.hub.load 加载本地模型?

我需要避免从网上下载模型(由于安装的机器的限制)。

这可行,但它从互联网下载模型

model = torch.hub.load('pytorch/vision:v0.9.0', 'deeplabv3_resnet101', pretrained=True)
Run Code Online (Sandbox Code Playgroud)

我已将.pth文件和hubconf.py文件放在 /tmp/ 文件夹中,并将我的代码更改为

model = torch.hub.load('/tmp/', 'deeplabv3_resnet101', pretrained=True, source='local')
Run Code Online (Sandbox Code Playgroud)

但令我惊讶的是,它仍然从互联网上下载模型。我究竟做错了什么?如何在本地加载模型?

只是为了向您提供更多详细信息,我在 Docker 容器中执行所有这些操作,该容器在运行时具有只读卷,因此这就是新文件下载失败的原因。

python machine-learning torch pytorch torchvision

11
推荐指数
2
解决办法
4万
查看次数

标签 统计

machine-learning ×1

python ×1

pytorch ×1

torch ×1

torchvision ×1