pytorch加载预训练模型报错-解决

最新推荐文章于 2023-08-02 10:51:54 发布
最新推荐文章于 2023-08-02 10:51:54 发布
阅读量2.4k 收藏 2
点赞数 1
分类专栏： python pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_40806289/article/details/106438549
版权

pytorch加载预训练模型时报错：

File "train_refinedet.py", line 115, in train
    refinedet_net.vgg.load_state_dict(vgg_weights)
  File "/home/em/anaconda3/lib/python3.5/site-packages/torch/nn/modules/module.py", line 769, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for ModuleList:
    Missing key(s) in state_dict: "0.bias", "0.

解决方法：

refinedet_net.vgg.load_state_dict(vgg_weights)
修改为：
refinedet_net.vgg.load_state_dict(vgg_weights，False)

 

逼疯代码
关注
专栏目录
预训练模型的加载机理pytorch版
认真努力，做一只会飞的毛毛虫。
02-25 6975
在做二维目标检测时，我们会对神经网络的部分地方进行修改，比如增加CBAM，或者修改FPN等等。但是将修改后的网络进行训练时，加载预训练权重过程却不会报错，甚至修改网络后性能不增反降。这里面蕴含的知识点在这篇文章做做一个总结。 pytorch中网络导入预训练权重的代码很简单： net = model() net.to(device) net.load_state_dict(torch.load('params.pth')) 其中： torch.load('params.pth')只是加载了模型参数，
使用pytorch-superpoint与pytorch-superglue项目实现训练自己的数据集
最新发布
a486259的博客
01-07 4589
superpoint与superglue的组合可以实现基于深度学习的图像配准，官方发布的superpoint与superglue模型均基于coco数据训练，与业务中的实际数据或许存在差距，为此实现基于开源的pytorch-superpoint与pytorch-superglue项目实现训练自己的数据集。然而，在训练pytorch-superpoint有诸多细节需要优化，特此整理成技术文档。本文档描述了实验pytorch-superpoint与pytorch-superglue项目训练自己的图像配准模型。
加载预训练模型时，原模型存在一些预训练模型没有的参数（以resnet为例）
qq_45347185的博客
05-18 1058
当前我们已经建立了模型的类resnet18，并且拥有预训练模型pth文件的地址 1. 查看模型权重 model_urls = { 'resnet18': 'https://s3.amazonaws.com/pytorch/models/resnet18-5c106cde.pth', 'resnet34': 'https://s3.amazonaws.com/pytorch/models/resnet34-333f7ec4.pth', 'resnet50': 'https://s3.a
pytorch加载预训练模型与自己模型不匹配的解决方法
找不到服务器的博客
02-27 6867
pytorch中如果自己搭建网络并且加载别人的与训练模型的话，如果模型和参数不严格匹配，就可能会出问题，接下来记录一下我的解决方法。 两个有序字典找不同 模型的参数和pth文件的参数都是有序字典(OrderedDict)，把字典中的键转为列表就可以在for循环里迭代找不同了。 model = ResNet18(1) model_dict1 = torch.load('resnet18.pth') model_dict2 = model.state_dict() model_list1 = list(mode
python模型训练报错_pycharm+keras+yolo3的使用和自选模型的训练中遇到的坑
weixin_35368330的博客
12-24 687
1、TensorFlow版本的问题报错：RuntimeError: `get_session` is not available when using TensorFlow 2.0.解决办法：这个问题是TensorFlow版本问题，在2.0以上get_session是被移除了。需要做一些修改，或者把tf降级。可以安装1.15版本：pip install tensorflow==1.15 --upg...
pytorch预训练、预训练时报错misssing key(s) in state_dict
sunflower_sara的机器学习园地
08-15 1869
使用： https://blog.csdn.net/Vivianyzw/article/details/81061765 1. 直接加载预训练模型 推荐方式： net.load_state_dict(torch.load("model.pth")) 2. 加载一部分预训练模型 见上面的链接 3. 微调经典网络 见上面的链接 4. 修改经典网络 见上面的链接 另外，对于...
PyTorch 源码解读之 torch.serialization & torch.hub
zandaoguang的博客
10-28 4197
作者 | 123456来源 | OpenMMLab编辑 | 极市平台导读本文解读基于PyTorch 1.7版本，对torch.serialization、torch.save和tor...
python怎么使用预训练的模型_PyTorch使用预训练模型
weixin_39567169的博客
11-21 3495
PyTorch模型加载的时候，有预训练模型，通过使用预训练模型可以给模型使用带来很多的便捷，对于模型的使用以下给出了一些总结，如有错误恳请指正。一、直接加载预训练模型进行训练1、加载保存的整个模型torch.save(model,'model.pkl')...model = torch.load('model.pkl')2、加载保存的模型参数torch.save(model.state_dict(...
pytorch预训练模型报错HTTPSConnectionPool(host='huggingface.co', port=443
10-08
这个错误通常是由于网络连接问题引起的。请确保你的网络连接正常，并检查是否可以访问 "huggingface.co" 的网站。...如果以上方法都无法解决问题，你可以考虑下载预训练模型的源文件，并手动加载到你的程序中。
预训练的Pytorch人脸检测（MTCNN）和识别（InceptionResnet）模型-Python开发
05-25
使用Pytorch Python 3.7 3.6 3.5状态的人脸识别这是pytorch中Inception Resnet（V1）模型的存储库，已在VGGFace2和CASIA上进行了预训练-我们使用Pytorch Python 3.7 3.6 3.5状态这是用于Inception Resnet（V1）模型...
程序发生run time error原因及解决方案
热门推荐
T_Y_F_的博客
08-07 2万+
程序发生run time error原因及解决方案
安装完操作系统之后，linux 启动失败，报错为"Invalid magic number 0 error 13: invalid or unsupported executable form......
chengluwei3014的博客
09-17 8010
安装完操作系统之后，linux 启动失败，报错为"Invalid magic number 0 error 13: invalid or unsupported executable format" 来源于： Oracle Linux: Boot Error "Invalid magic n...
执行model.load_state_dict报错map_location=torch.device(‘cpu‘)以及Unexpected key(s) in state_dict
m0_56222998的博客
08-02 1928
本蒟蒻原先是在有GPU的服务器上训练得到了模型，现在想在没有GPU的服务器部署网页，需要加载该模型，当运行了下面代码时。是因为没有将state_dict转移到CPU上，在PyTorch中，可以通过指定。仔细查看提示信息，发现是有一个未知的Key，我把它删掉竟然就跑通了。我打印出了两台服务器中state_dict的内容。打印的内容都是一样的，都是下面的内容。
attempt_load() got an unexpected keyword argument ‘map_location‘
AI浩
11-02 6417
yolov5的版本做了升级了。attempt_load的参数做了调整。将map_location改为device。
pytorch加载下载好的预训练模型出现问题
桐原因的博客
11-13 9383
在官网上下载好pytorch版的预训练模型，
python个人bug修炼手册
dusk_and_night的博客
07-19 9845
python、pytorch、bug、pip
model.load_state_dict(state_dict)报错问题
xuru_0927的博客
07-31 1万+
看一下训练得到的模型参数 state_dict = torch.load('logs/sanity_3/checkpoint', map_location='cuda' if args['train']['cuda'] else 'cpu') state_dict = state_dict['model'] 看一下参数 for k,v in state_dict.items(): print(k) 输出： module.block.0.layers.0.weight modul.
【创新实训】风格迁移功能探索与实现（四） train_model.py 训练代码的编写
cym的博客
05-30 992
最短路径问题Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 33546    Accepted Submission(s): 9840Problem Description给你n个点，m条无向边，每条边都有长度d和花费p，给你起点s终点t，...
pytorch快速上手（5）-----pytorch模型的保存加载与断点恢复训练
All_In_gzx_cc的博客
02-02 1206
模型的保存与加载 PyTorch中的保存（序列化，从内存到硬盘）与反序列化（加载，从硬盘到内存） torch.save主要参数： obj：对象 、f：输出路径 torch.load 主要参数 ：f：文件路径 、map_location：指定存放位置、 cpu or gpu 一、常见的模型保存的两种方法： 1、保存整个Module torch.save(net, path) 2、只保存模型参数 state_dict = net.state_dict() torch.save(state_dict , p
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值