torch.hub.load改为本地
时间: 2023-04-23 14:04:54 浏览: 334
基于Torch Hub的深度估计模型MiDaS-python源码.zip
torch.hub.load可以从远程仓库加载预训练模型,如果要改为本地加载,可以将预训练模型下载到本地,然后使用torch.load加载模型文件。具体步骤如下:
1. 下载预训练模型文件到本地,例如保存为'model.pth'。
2. 使用torch.load加载模型文件,例如:
```
import torch
model = torch.load('model.pth')
```
3. 使用加载的模型进行推理或微调等操作。
阅读全文