Pytorch中更改预训练权重文件的下载位置

1. 参考链接

• Pytorch更改预训练权重下载位置
• pytorch———修改预训练模型下载路径

2. 更改方法

在线加载的预训练权重默认存放位置在：C:\Users\[用户名]\.cache\torch\hub\checkpoints

方法1：在某个.py文件最上方加上以下代码

import os
# 更改预训练下载位置
os.environ['TORCH_HOME'] = '/data/Your_User_Name/pretrained_weights/'

方法2：更改model_dir，以后所有预训练权重就直接下载到指定的model_dir中了（不管是AlexNet还是vgg16或者其他模型的预训练权重都会直接下载到model_dir中，不用再像方法1那样每次去手动设置下载位置了）。

step1：Ctrl+左键，进入vgg16

step2：Ctrl+左键，进入_vgg

step3：Ctrl+左键，进入load_state_dict_from_url

step4：更改model_dir，例如model_dir='D:/torch-pretrained'

这里要注意一下文件路径中\和/的使用。

• 在python文件中使用/来表示路径，\表示转义。

• 但是很多情况下在Windows下直接复制文件夹的地址（例如下图）是用\表示的。

假设待设置的路径是D盘下的torch-pretrained文件夹，则有以下3种书写方式：

• model_dir='D:/torch-pretrained'，直接全部写为/
• model_dir='D:\\torch-pretrained'，写成\\
• model_dir=r'D:\torch-pretrained'，r和\配合使用

3. 一个小技巧

执行程序，发现pth文件下载速度很慢，此时我们可以用迅雷下载，如图所示，再将文件放入我们修改的路径中即可。

下载地址在vgg.py文件中有：

model_urls = {
    'vgg11': 'https://download.pytorch.org/models/vgg11-bbd30ac9.pth',
    'vgg13': 'https://download.pytorch.org/models/vgg13-c768596a.pth',
    'vgg16': 'https://download.pytorch.org/models/vgg16-397923af.pth',
    'vgg19': 'https://download.pytorch.org/models/vgg19-dcbb9e9d.pth',
    'vgg11_bn': 'https://download.pytorch.org/models/vgg11_bn-6002323d.pth',
    'vgg13_bn': 'https://download.pytorch.org/models/vgg13_bn-abd245e5.pth',
    'vgg16_bn': 'https://download.pytorch.org/models/vgg16_bn-6c64b313.pth',
    'vgg19_bn': 'https://download.pytorch.org/models/vgg19_bn-c79401a0.pth',
}

先操作方法2的step1、step2，再如下图所示，即可看到上面那些下载地址：