pytorch———修改预训练模型下载路径

首次用Pytorch加载预训练模型，需要在线下载，默认地址在C盘。

并且，我们可在本地查看到该文件已被创建，

随着模型的增加，我们的C盘空间会越来越小。并且，本人不太喜欢在C盘中存放东西，因此，更改下载的默认路径很重要。

下面就介绍两种常见的路径修改的方法。

方法1：

下面以resnet50为例

1、Ctrl + 鼠标左键，查找自己代码里所调用网络的类,查看此网络的加载方法，如图所示：

2、Ctrl + 鼠标左键，继续查找，如图所示：

3、model_dir便是我们模型存储的地址，默认值为None。因此，我们将model_dir修改成我们模型存储的路径即可。

博主的地址为：（仅供参考）

4、修改后，我们再次执行程序，则发现，我们的模型存储的路径变为我们设置的路径了，如图所示：

并且在本地也可看到该文件，

方法2：

以vgg19为例

1、建立一个model文件夹，并在该文件夹下建立__init__.py和vgg.py文件，文件结构如图，

2、Ctrl + 鼠标左键，找到vgg19，如图所示，

3、Ctrl + A，将其复制粘贴到我们创建的models/vgg.py中，如图所示，

4、在models/__init__.py文件中添加from .vgg import vgg19

5、在models/vgg.py文件中的vgg19()函数中添加model_dir = 'E:/checkpoint'(此处为模型存储的路径，可根据个人喜好修改)

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

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

以上就是更改路径的方法。