Pytorch修改内置模型下载路径（vgg等）

最新推荐文章于 2024-05-14 11:48:57 发布

NorthSmile

最新推荐文章于 2024-05-14 11:48:57 发布

阅读量2k

点赞数 5

分类专栏：深度学习文章标签： pytorch 深度学习人工智能

本文链接：https://blog.csdn.net/qq_43665602/article/details/126327094

版权

深度学习同时被 3 个专栏收录

37 篇文章 82 订阅 ¥9.90 ¥99.00

订阅专栏

超级会员免费看

Pytorch

32 篇文章 12 订阅

订阅专栏

Python

8 篇文章 0 订阅

订阅专栏

修改内置模型下载路径

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

一、前言：

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

二、举例：

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

三、修改步骤：

1）将光标移至vgg16，Ctrl+左击进入vgg16的实现源码；在这里插入图片描述
2）找到vgg.py文件导包部分的load_state_dict_from_url，对其Ctrl+左击进行定位；

3）自定义模型下载路径，并对model_dir进行替换，如图所示：