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

孟孟单单

已于 2022-05-16 17:26:29 修改

阅读量7.2k

点赞数 11

分类专栏：深度学习项目经验tips 文章标签： pytorch 深度学习

于 2022-05-16 12:17:21 首次发布

本文链接：https://blog.csdn.net/LWD19981223/article/details/124796466

版权

深度学习项目经验tips 专栏收录该内容

40 篇文章

订阅专栏

1. 参考链接

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',
}