有什么办法可以将 PyTorch 中可用的预训练模型下载到特定路径？

Question

我指的是可以在这里找到的模型： https ://pytorch.org/docs/stable/torchvision/models.html#torchvision-models

原文由 gopalkrizna 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

因为， @dennlinger 在他的 回答 中提到： torch.utils.model_zoo ，在您加载预训练模型时被内部调用。

更具体地说，每次加载预训练模型时都会调用方法： torch.utils.model_zoo.load_url() 。同样的文档提到：

model_dir $TORCH_HOME $TORCH_HOME/models ~/.torch

可以使用 $TORCH_HOME 环境变量覆盖默认目录。

这可以按如下方式完成：

 import torch
import torchvision
import os

# Suppose you are trying to load pre-trained resnet model in directory- models\resnet

os.environ['TORCH_HOME'] = 'models\\resnet' #setting the environment variable
resnet = torchvision.models.resnet18(pretrained=True)

我通过在 PyTorch 的 GitHub 存储库中提出问题来遇到上述解决方案： https ://github.com/pytorch/vision/issues/616

这导致了文档的改进，即上述解决方案。

原文由 gopalkrizna 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

是的，您可以简单地复制网址并使用 wget 将其下载到所需路径。这是一个例子：

对于 亚历克斯网：

 $ wget -c https://download.pytorch.org/models/alexnet-owt-4df8aa71.pth

对于 Google Inception (v3) ：

 $ wget -c https://download.pytorch.org/models/inception_v3_google-1a9a5a14.pth

对于 挤压网：

 $ wget -c https://download.pytorch.org/models/squeezenet1_1-f364aa15.pth

对于 MobileNetV2 ：

 $ wget -c https://download.pytorch.org/models/mobilenet_v2-b0353104.pth

对于 DenseNet201 ：

 $ wget -c https://download.pytorch.org/models/densenet201-c1103571.pth

对于 MNASNet1_0 ：

 $ wget -c https://download.pytorch.org/models/mnasnet1.0_top1_73.512-f206786ef8.pth

对于 ShuffleNetv2_x1.0 ：

 $ wget -c https://download.pytorch.org/models/shufflenetv2_x1-5666bf0f80.pth

---

如果你想用 Python 来做，那么使用类似的东西：

 In [11]: from six.moves import urllib

# resnet 101 host url
In [12]: url = "https://download.pytorch.org/models/resnet101-5d3b4d8f.pth"

# download and rename the file to `resnet_101.pth`
In [13]: urllib.request.urlretrieve(url, "resnet_101.pth")
Out[13]: ('resnet_101.pth', <http.client.HTTPMessage at 0x7f7fd7f53438>)

PS：在 torchvision.models 的各个python模块中可以找到下载地址

原文由 kmario23 发布，翻译遵循 CC BY-SA 4.0 许可协议