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

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

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

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

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

方法1:

下面以resnet50为例

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

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

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

博主的地址为:(仅供参考)

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

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

方法2:

vgg19为例

1、建立一个model文件夹,并在该文件夹下建立__init__.pyvgg.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文件中有,

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

猜你喜欢

转载自blog.csdn.net/weixin_45666660/article/details/108710898