加载模型出现 RuntimeError: Error(s) in loading state_dict for Model: Missing key(s) in state_dict

最新推荐文章于 2024-12-20 09:53:13 发布
最新推荐文章于 2024-12-20 09:53:13 发布
阅读量828 收藏 2
点赞数 4
分类专栏： 图像增强 文章标签： 计算机视觉
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/DQHNB/article/details/143105537
版权

在这里插入图片描述
原因：加载使用模型时和训练模型时的环境不一致

解决办法
将load_state_dict(state_dict) 改成 model.load_state_dict(state_dict, False)

load_state_dict方法参数的官方说明 strict 参数默认是true，他的含义是 是否严格要求state_dict中的键与该模块的键返回的键匹配
在这里插入图片描述
当strict设置为False，模型加载状态字典时将不会严格检查每个键是否匹配，允许部分加载，这在处理预训练模型或自定义模块时非常有用

RuntimeError: Error(s) in loading state_dict for ResNet: Unexpected key(s) in state_dict: “bn1.num
优快云 精品推荐
01-09 1703
在这个特定的情况下，错误信息中提到了多个“num_batches_tracked”键。这类键在某些 PyTorch 版本中存在，而在其他版本中可能不存在。方法会加载所有匹配的键，而忽略任何不匹配的键，从而避免了这个错误。但是，这种方法应谨慎使用，因为它可能掩盖模型结构和状态字典之间的不一致问题。这样做会使得 PyTorch 忽略不匹配的键，只加载那些与模型结构相匹配的键。这通常发生在模型的结构和要加载的状态字典不完全匹配时。解决这个问题的一个常见方法是在加载状态字典之前，明确地忽略不匹配的键。
RuntimeError: Error(s) in loading state_dict for DistributedDataParallel: Missing key(s) in state_di
weixin_57948968的博客
05-05 645
这个错误通常是因为键值对不匹配的问题，也就是说在训练过程中，你保存的checkpoints文件中的键值对对应不上，这个时候你需要把原来的训练的文件删除，然后再重新进行训练即可。
pytorch加载模型错误 RuntimeError: Error(s) in loading state_dict for Model: Missing key(s) in state_dict
edward_zcl的博客
04-28 3570
模型在保存时侯以键对值保存，同时在加载时根据现在网络的键值查找模型对应的键值，然后加载。一般报错是因为模型和网络的键值不匹配。 1、最常见的问题是键值多了或者少了 module. 此种情况是模型在DataParallel或者DDP训练后保存的键值有module. ，对应的网络的键值则没有module. 1)可以通过： model = nn.DataParallel(model) 将模型的键值...
已解决：RuntimeError: Error(s) in loading state_dict for U2Net
qq_53354720的博客
08-14 490
这是权重文件中的某些层的数据类型（如浮点数精度）与模型中的对应层不一致 导致的。解决办法：在加载权重时使用。参数来忽略不匹配的层。
深度学习避坑提示：RuntimeError: Error(s) in loading state_dict for Model: Missing key(s) in state_dict
tuanzi2809的博客
02-19 1万+
RuntimeError: Error(s) in loading state_dict for Model: Missing key(s) in state_dict
RuntimeError: Error(s) in loading state_dict for Model: size mismatch for model.24.m.0.weight....
qq_44442727的博客
11-07 2570
但是，这处的问题并不好改，如果也是使用yolov5_rotated代码训练，建议对照yolov5的训练代码，可以发现这个rotated框架在优化器部分的代码写的不完善，而且预训练判断部分的代码也有问题，即使成功训练起来，epoch数量也不是从0，1开始，所以主要还是框架考虑不够完善。查找了一些办法，基本上是关于 loading_state_dict()的用法，如何去处理加载的权重，开始直接定位到了ckpt[‘model’]的这几层的权重，强行将几层的权重维度匹配到model中，但是训练时仍然没有解决问题。
PyTorch加载模型出现Error(s) in loading state_dict() for Model问题，Unexpected key(s) in state_dict: “...“
热门推荐
Albert的博客
11-04 3万+
问题： 模型在训练过程中可以正常训练，但是测试的时候出现了错误，如下所示： RuntimeError: Error(s) in loading state_dict for ModuleList: Missing key(s) in state_dict: "0.weight", "1.weight", "1.bias", "1.running_mean", "1.running_var", Unexpected key(s) in state_dict: "conv1.weight", "bn1.we
赛博丹炉训练问题RuntimeError: Error(s) in loading state_dict for UNet2DConditionModel:Missing key(s)...
weixin_57625762的博客
08-05 841
新手小白在优快云上的第一篇记录，还望大家多多鼓励。今天在使用赛博丹炉训练lora时，开始训练出现报错提醒：RuntimeError: Error(s) in loading state_dict for UNet2DConditionModel:Missing key(s) in state_dict:,记录一下解决过程与结果。
解决 RuntimeError: Error(s) in loading state_dict for : Missing key(s) in state_dict
lei_qi的博客
07-09 1万+
使用torch.nn.DataParallel多卡训练模型之后，加载模型前也需要打开多卡读取模型。 我最近使用多卡训练了一个模型。保存的方式是state_dict的方式。 然后在加载模型的时候就一直出错。 raise RuntimeError(‘Error(s) in loading state_dict for {}:\n\t{}’.format( RuntimeError: Error(s) in loading state_dict for BertMultiClassification: Missi
模型加载出现问题，RuntimeError: Error(s) in loading state_dict for UNet：Missing key(s) in state_dict...
qq_53763141的博客
03-07 1368
这是因为用DataParallel训练的模型数据并行方式的，key中会包含”module“关键字。（2）去掉DataParallel 预训练model中的module(可行权重值一致)
RuntimeError: Error(s) in loading state_dict for
sgfsfgs的博客
03-06 1929
版本问题不对应，降低pyopengl 版本 至3.1.4。
（已解决）RuntimeError: Error(s) in loading state_dict for Generator: Missing key(s) in state_dict: “hpf
qq_45937473的博客
03-13 2904
这个错误表明在加载模型的状态字典时出现了问题，因为状态字典中缺少了一个名为"hpf.filter"的键。这通常是由于尝试加载的模型与代码中期望的模型结构不匹配所导致的。可能原因包括：模型版本不匹配，模型参数丢失或者损坏。加了一个False，加这个的原因是：是否严格匹配状态字典中的键和模型中的模块。如果设置为False，则可以忽略不匹配的键，只加载匹配的部分。在我们使用PyTorch中用于加载模型状态字典（state_dict）的方法调用。目前在复现一篇论文的代码，执行的时候报错。
【Error】Pytorch测试神经网络时出现“RuntimeError: Error(s) in loading state_dict for *** “
⛄
01-24 779
解决方法一： load_state_dict(torch.load(‘net.pth’)在前，增加 model = nn.DataParallel(model) 就可以了。 例如： net = NET() net.cuda() net = nn.DataParallel(net) net.load_state_dict(torch.load('net.pth') 如果还不行可以考虑是pytorch版本换成大于1.0.0（小于0.4.0），若果是别人训练好的权重还应该考虑是不是自己的模型和别人的权重文件不匹
RuntimeError: Error(s) in loading state_dict for Dual_net: Missing key(s) in state_dict:
最新发布
qq_55795524的博客
12-20 214
问题：模型是对的，但是还是报错。
加载模型出现-RuntimeError: Error(s) in loading state_dict for Net:unexpected key(s) in state_dict: XXX
与君共勉，一起学习
05-04 7553
加载模型时发生错误RuntimeError: Error(s) in loading state_dict for Net:unexpected key(s) in state_dict: XXX Traceback (most recent call last): File "demo.py", line 380, in <module> model.load_state_dict(torch.load('./0428.pth')) File "/root/miniconda
RuntimeError: Error(s) in loading state_dict for AutoEncoderModel......
aixiaomi123的博客
07-05 1172
训练完model1之后保存了模型参数，命名为checkpoint_1；然后在model_1的基础上修改了模型，在内部添加了一些新的参数（或者更改了参数的名称），命名为model_2，训练之后保存了模型参数，命名为checkpoint_2。报错：RuntimeError: Error(s) in loading state_dict for Resnet50 Unexpected key(s) in state_dict:““在训练时可能会出现以上错误：原因是由于加载错误的模型参数导致的。
RuntimeError: Error(s) in loading state_dict for (Missing key(s) in state_dict: "firstconv.weight)
qq_20373723的博客
01-09 3740
错误记录太长了，解决在最下面 D:\xxx\xxx\complete_project\xxxxx>python3 predict.py Traceback (most recent call last): File "predict.py", line 220, in <module> test_ce_net_vessel() File "predict.py"...
RuntimeError: Error(s) in loading state_dict for ConvNeXt: Missing key(s) in state_dict:
02-01
这个错误通常是由于加载模型时，模型的state_dict中缺少了一些键值对导致的。state_dict是一个字典对象，它将每个模型参数的名称映射到其对应的参数张量。当加载模型时，PyTorch会尝试将保存的state_dict加载到模型...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值