我需要避免从网上下载模型(由于安装的机器的限制)。
这可行,但它从互联网下载模型
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 容器中执行所有这些操作,该容器在运行时具有只读卷,因此这就是新文件下载失败的原因。