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

Question

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

Answer 1

正如，@dennlinger在他提到的回答：torch.utils.model_zoo，是当你加载一个预先训练模型被内部调用。

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

默认值model_dir是$TORCH_HOME/models其中 $TORCH_HOME默认~/.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

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

Answer 2

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

对于AlexNet：

$ 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

对于SqueezeNet：

$ 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模块中找到