【Pytorch】加载torchvision中预训练好的模型并修改默认下载路径（使用models.__dict__[model_name]()读取）

最新推荐文章于 2025-10-10 18:45:58 发布
原创 最新推荐文章于 2025-10-10 18:45:58 发布 · 1.1w 阅读 · 60 ·
CC 4.0 BY-SA版权
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
文章标签：

#python #pytorch #深度学习 #经验分享

本文介绍了如何在Windows10环境下，使用Anaconda的Python环境中，修改PyTorch torchvision加载预训练模型时的默认下载路径。通过查找源码并修改相应函数，实现模型保存在指定目录下，以方便统一管理。

说明

使用torchvision.model加载预训练好的模型时，发现默认下载路径在系统盘下面的用户目录下（这个你执行的时候就会发现），即C:\用户名\.cache\torch\.checkpoints下，便于统一管理，我决定修改model的存放路径，在网上找了很久都没有很好的解决方法，只能自己尝试，现将解决方案给出，供大家参考~

操作环境

  • windows10 + Anaconda
  • torch:1.1.0
  • torchvision:0.3.0

加载方式

以加载vgg16为例，首先定义网络结构

class CNN(nn.Module):

    def __init__(self, usegpu=True)
最低0.47元/天 解锁文章
Pytorch修改内置模型下载路径（vgg等）
NorthSmile的博客
08-14 2701
在使用Pytorch内置的模型时，会默认下载到用户家目录下，而对于很多人来说并不想在C盘下载一些较大的文件，此时修改模型的下载路径则显得比较重要。
如何利用torchvision.models已有的模型训练自己的分类数据集
qq_53536373的博客
04-09 656
scheduler = optim.lr_scheduler.StepLR(optimizer_ft, step_size=10, gamma=0.1)#学习率每7个epoch衰减成原来的1/10。optimizer_ft = optim.Adam(params_to_update, lr=1e-2)#要训练啥参数，你来定。model_ft.fc = nn.Linear(num_ftrs, 102) # 类别数自己根据自己任务来。input_size = 64 # 输入大小根据自己配置来。
【Torchvision】torchvision.models 模块：预定义的计算机视觉模型
最新发布
彬彬侠的博客
10-10 1545
torchvision.models 是 PyTorch 视觉库的核心模块，提供丰富的预训练计算机视觉模型，包括图像分类（ResNet、ViT）、目标检测（Faster R-CNN）、图像分割（DeepLabV3）、视频分类（R3D）和量化模型。支持迁移学习，可冻结特征提取层或微调整个网络，并兼容 PyTorch 生态。通过简单接口加载模型（如 resnet50(weights='IMAGENET1K_V1')），支持自定义分类头以适应不同任务。该模块简化了模型开发流程，适用于研究和生产环境。
利用torchvision.models调用现成的网络
weixin_44586881的博客
04-11 7106
以resnet50为例，其最简单的调用方式就是：↓不需要初始化什么参数，这样得到的model就是默认的resnet50结构，可以直接用来做分类训练。但是还提供了预训练参数权重，只需要：↓这种方式会直接从官网上进行预训练权重的下载，该预训练权重是由ImageNet-1K（标准输入224x224）而来，由于其本质是一个分类网络，所以最后的全连接层大小为1000.但是这种方式会被警告，因为pretrained参数可能以后会被舍弃，建议我们使用weight参数。这样的话我们有两种方法。
PyTorch源码解读（四）torchvision.models
sinat_42239797的博客
07-02 3344
PyTorch框架中有一个非常重要且好用的包：torchvision，该包主要由3个子包组成，分别是：torchvision.datasets torchvision.models、torchvision.transforms。这3个子包的具体介绍可以参考官网https://pytorch.org/docs/master/torchvision/datasets.html。 我的另外两篇博客对其...
PyTorch源码解读之torchvision.models
热门推荐
AI之路
01-21 10万+
PyTorch框架中有一个非常重要且好用的包：torchvision，该包主要由3个子包组成，分别是：torchvision.datasets、torchvision.models、torchvision.transforms。这3个子包的具体介绍可以参考官网：http://pytorch.org/docs/master/torchvision/index.html。具体代码可以参考github：
修改PyTorch预训练模型的下载路径
yanxiangtianji的专栏
01-06 1万+
PyTorch可以下载很多预训练好的神经网络，用起来会很方便。但是默认的下载路径是用户根目录下的.cache文件夹。在Linux和Mac下还比较符合用户习惯，但是在windows下就是C:\Users\<username>\.cache。不仅不符合很多windows用户的使用习惯而且不利于文件的保存。另外如果是多人共享的Linux服务器，每个用户都下载一遍这些训练好的神经网络明显是一种浪费。最好是放置在/var、/opt或者/usr/local等一类共享目录里。 但是默认情况下，PyTorch里
model = torchvision.models.__dict__[name](pretrained=True)
03-14
好的，我现在需要解决用户关于如何使用torchvision.models模块中的__dict__方法加载预训练模型并设置pretrained=True的问题。首先，我得回忆一下相关的PyTorch知识。用户提到引用了三个资料，其中第一个引用提到了...
python torchvision_Python torchvision.models 模块，__dict__() 实例源码 - 编程字典
weixin_39542514的博客
12-21 609
def _nn_forward_hook(self, module, input, output, name=''):if type(output) is list:self.blobs[name] = [o.data.clone() for o in output]else:self.blobs[name] = output.data.clone()# @staticmethod# def _l...
model.load_state_dict(checkpoint[‘model_state_dict‘]) File “E:\666\lib\site-packages\torch\nn\
sinat_38424109的博客
11-18 631
加载预训练模型: E:\Pointnet_Pointnet2_pytorch\log\classification\pointnet2_msg_normals\checkpoints\best_model.pth。虚拟环境 666 已激活。设备设置为: cpu。
Pytorch_使torchvision只从本地加载代码，不再自动下载模型
thiefun2020的博客
09-15 1242
最近在windows系统下使用pytorch1.5.0和torchvision0.6.1做一个小项目（使用torchvision内置的fasterrcnn模型进行目标检测）的时候发现一个小问题，即torchversion会自动下载一个resnet50的模型到一个系统路径：C:\Users\**你的windows用户名**\.cache\torch\checkpoints下，这种内置方法既不方便（模型的下载源在国外，下载比较慢），也不好部署，网上也查到一些方法，包括：pytorch预训练模型的下载地址以及解决
pytorch加载torchvision中预训练好的模型时，更改模型保存地址
tt丫的博客
02-25 3331
入门小菜鸟，希望像做笔记记录自己学的东西，也希望能帮助到同样入门的人，更希望大佬们帮忙纠错啦~侵权立删。 感谢【Pytorch】加载torchvision中预训练好的模型并修改默认下载路径（使用models.__dict__[model_name]()读取） - 灰信网（软件开发博客聚合） 今天我在执行这行代码时，它开始给我把resnet101的预训练模型下载安装在我的C盘里！！！ model = getattr(models, backbone)(pretrained) 可是我的C盘已经要这
pytorch加载预训练模型
weixin_45807767的博客
03-09 1877
加载预训练模型 1/4 直接加载预训练模型 import torchvison.models as models resnet50 = models.resnet50(pretrained=True) # 若只需要网络结构，不需要预训练模型的参数来初始化，则设置pretrained=False，或者把resnet复制到自己的目录下，新建个model文件夹，可以参考下面的猫狗大战入门算法入门：https://github.com/JackwithWilshere/Kaggle-Dogs_vs_Cats_PyT
TorchVision的使用方法、更改默认路径
weixin_44144773的博客
02-03 1598
TorchVision的使用方法、更改默认路径、数据增强、加载模型、修改模型默认下载位置、models
加载torchvision中预训练好的模型并修改默认下载路径
sueong的博客
09-29 483
原因是home目录下的硬盘满了 所以只好改pytorch下载预训练的模型 参考https://blog.csdn.net/ProLover98/article/details/104792115 成功！！
TORCHVISION MODELS
Claroja
08-24 400
随机weights创建model import torchvision.models as models resnet18 = models.resnet18() alexnet = models.alexnet() vgg16 = models.vgg16() squeezenet = models.squeezenet1_0() densenet = models.densenet161() inception = models.inception_v3() googlenet = models.goo
学习torchvision.models._dict_
lbling123的博客
05-26 6318
torchvision.models._dict_[arch](pretrained=False)与python _dict_ 的关系   之前一直在写面向过程的C代码，转换到Python面向对象了有点不适应，前几天学习pytorch的时候，训练模型使用torchvision.models模块来加载预训练模型，记住了一种调用方法，如下： model = torchvison.models.__dict__[arch](pretrained=True) //arch是需要加载的预训练模型名，比如‘resnet
【PyTorch】Torchvision Models
leoatliang的博客
11-24 1850
Torchvision Models
更改torchvision预训练模型保存路径
weixin_45406060的博客
10-21 682
更改torchvision已训练模型保存路径
评论 11
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
表情包 代码片
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值