Pytorch修改内置模型下载路径(vgg等)

修改内置模型下载路径

  • 一、前言:
  • 二、举例:
  • 三、修改步骤:

一、前言:

在使用Pytorch内置的模型时,会默认下载到用户家目录下,
而对于很多人来说并不想在C盘下载一些较大的文件,此时修改模型的下载路径则显得比较重要。

二、举例:

使用预训练好的vgg16
在这里插入图片描述

三、修改步骤:

1)将光标移至vgg16,Ctrl+左击进入vgg16的实现源码;在这里插入图片描述
2)找到vgg.py文件导包部分的load_state_dict_from_url,对其Ctrl+左击进行定位;
Pytorch修改内置模型下载路径(vgg等)_第1张图片
3)自定义模型下载路径,并对model_dir进行替换,如图所示:
在这里插入图片描述
4)此时可看到模型的下载路径已经更改为自定义路径:在这里插入图片描述

你可能感兴趣的:(Pytorch,Python,深度学习,pytorch,深度学习,人工智能)