【pytorch】加载模型出现问题 Unexpected key(s) in state_dict: “params“, “params_ema“

最新推荐文章于 2023-12-12 15:48:18 发布
最新推荐文章于 2023-12-12 15:48:18 发布
阅读量1.1k 收藏 1
点赞数
文章标签： pytorch 深度学习 人工智能
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/Coco_W/article/details/129961935
版权

最开始

model.load_state_dict(torch.load(model_path), strict=True)

加载模型报错如下

Missing key(s) in state_dict: "conv_first.weight", "conv_first.bias", .... 
Unexpected key(s) in state_dict: "params", "params_ema"

训练好的模型键值有许多，只需要模型参数文件，我的参数文件在params_ema内
修改如下即可

state_dict = torch.load(model_path)
model.load_state_dict(state_dict['params_ema'], strict=True)
今天也要好好生活
关注
  • 0
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
pytorch加载模型报错Unexpected key(s) in state_dict: module.conv1.weight, module.bn1
Hello Word!
06-28 1万+
文章目录代码报错原因解决 代码 import models arch = 'resnet50' model = models.__dict__[arch]() checkpoint = torch.load(ckptFile) model.load_state_dict(checkpoint['state_dict']) model = torch.nn.DataParallel(model).cuda() 报错 Traceback (most recent call last
RuntimeError: Error(s) in loading state_dict for ResNet: Unexpected key(s) in state_dict: “bn1.num
最新发布
CSDN 精品推荐
01-09 764
在这个特定的情况下，错误信息中提到了多个“num_batches_tracked”键。这类键在某些 PyTorch 版本中存在，而在其他版本中可能不存在。方法会加载所有匹配的键，而忽略任何不匹配的键，从而避免了这个错误。但是，这种方法应谨慎使用，因为它可能掩盖模型结构和状态字典之间的不一致问题。这样做会使得 PyTorch 忽略不匹配的键，只加载那些与模型结构相匹配的键。这通常发生在模型的结构和要加载的状态字典不完全匹配时。解决这个问题的一个常见方法是在加载状态字典之前，明确地忽略不匹配的键。
pytorch使用torch.load加载模型出现错误Unexpected key(s) in state_dict
风影的博客
09-04 3312
记录错误积累经验 猛然来这么一大串错误信息让我有点震惊。 pytorch模型文件以key，value形式保存，现在模型所有key都对不上，有点奇怪。细看发现加载的模型相较于当前模型其key值前面多了“module."。 初步诊断 class Tester(object): TestParams = TestParams def __init__(self, model, test_params): assert isinstance(test_params, TestP
Unexpected key(s) in state_dict: “model“
weiyuangong的博客
05-26 1474
问题描述：加载预训练权重时报错Unexpected key(s) in state_dict: "model" weight_dic_path = 'convnext_base_1k_224.pth' state_dic=torch.load(weight_dic_path,map_location=device) model.load_state_dict(state_dic) 解决方法： weight_dic_path = 'convnext_base_1k_224.pth' state_dic
PyTorch加载模型model.load_state_dict()问题，Unexpected key(s) in state_dict: “module.features..，Expected .
热门推荐
是否龍磊磊真的一无所有的博客
04-17 6万+
希望将训练好的模型加载到新的网络上。如上面题目所描述的，PyTorch在加载之前保存的模型参数的时候，遇到了问题。 Unexpected key(s) in state_dict: "module.features. ...".，Expected ".features....". 直接原因是key值名字不对应。 表明了加载过程中，期望获得的key值为feature...，而不是...
解决torch.load中Unexpected key(s) in state_dict的报错问题
weixin_51007334的博客
08-22 1670
今天在使用强化学习跑实验的时候，使用到了这个函数，从已经保存的pth文件中加载模型权重。但是加载的过程中，出现了错误。
Unexpected key(s) in state_dict: “module.backbone.bn1.num_batches_tracked“
牛肉胡辣汤
11-06 1642
当加载保存的​​时，出现​​错误通常是由于键名不匹配引起的。我们可以通过查看模型的键名和保存的​​的键名来找出不匹配的键，并相应地修改它们。另外，使用​​包装模型可以解决多GPU训练导致的键名前缀问题。希望本文能帮助你解决​​错误，并顺利加载模型参数。
pytorch 预训练model加载问题”Unexpected key(s) in state_dict:“
qq_23116521的博客
03-02 1万+
预训练模型使用 model = ResNet50() model = nn.DataParallel(model).cuda() 加载预训练模型是使用 model = ResNet50().cuda() checkpoint = torch.load(args.net_cache) model.load_state_dict(checkpoint['state_...
pytorch加载模型报错Missing key(s) in state_dict: “classifier00.0.fc.weight“, “classifier00.0.fc.bias“
Hello Word!
07-08 4197
报错： Traceback (most recent call last): File "/home/user1/load_model_test.py", line 23, in <module> learner.load_state(conf, 'ir_se50.pth', model_only=True, from_save_folder=True) File "/home/user1/Learner.py", line 84, in load_state self.
ValueError: (‘Unexpected parameters in params‘, dict_keys([‘cv‘]))
data+scenario+science+insight
10-11 1439
ValueError: ('Unexpected parameters in params', dict_keys(['cv']))
Missing key(s) in state_dict | unexpected_keys | 模型加载 [torch.load] 报错【解决方案】
专注于AI领域前沿技术学习与分享：目标检测、图像修复、超分重建、AI工程化
09-24 1万+
文章目录Missing key(s) in state_dict 报错分析该报错快速直接解决方法额外思路【Unexpected key(s) in state_dict】- 解决思路 Missing key(s) in state_dict 报错分析 在执行 yolov3 pytorch 模型加载时遇到如下报错 类似报错之前其实也遇到过,今天总结一下 所运行代码 # Initialize model model = Darknet(opt.cfg, imgsz) #
Missing key(s) in state_dict: Unexpected key(s) in state_dict 两种解决方法 祝你成功！！！
weixin_61075104的博客
02-20 1811
Missing key(s) in state_dict: Unexpected key(s) in state_dict
“torch.load“中出现的“Unexpected key(s) in state_dict“报错问题
weixin_43723625的博客
10-17 2702
允许加载过程中出现不匹配的键。但请注意,仍然需要确保模型中的主要参数能够正确加载，以确保模型的有效性。
yolov8 + EMA 出现的问题：KeyError: ‘EMA‘
qq_45640098的博客
12-12 848
KeyError: 'EMA'解决方法
Pytorch：Unexpected key(s) in state_dict:
wchzh2015的博客
12-25 1万+
实验环境 pytorch 0.4 Python 3.6 在使用交叉验证后，将保存的模型重新加载求平均的结果。但是出现了以下的错误。 实验的bug截图 在此感谢这篇博文：https://blog.csdn.net/jacke121/article/details/82495184 自己修改后的关键代码如下： model_resnet101 = get_net...
成功解决 params, model_params = config_reader()config_reader KeyError: ‘param‘
近期请国内外头部出版社可尽快私信博主！——心比天高，仗剑走天涯，保持热爱，奔赴向梦想！低调，谦虚，自律，反思，成长，还算是比较正能量的博主，公益免费传播……内心特别想在AI界做出一些可以推进历史进程影响力的东西(兴趣使然，有点小情怀，也有点使命感呀)…
12-01 2662
成功解决 params, model_params = config_reader()config_reader KeyError: 'param' 目录 解决问题 解决思路 解决方法 解决问题 params, model_params = config_reader() config_reader param = config['param'] File "E:\Program Files\Python\Python36\lib\site-pac...
GFPGAN源代码分析（六）
qq_45969525的博客
12-27 1341
2021SC@SDUSC 一、分析的代码片段 1.代码展示 if different_w: linear_out_channel = (int(math.log(out_size, 2)) * 2 - 2) * num_style_feat else: linear_out_channel = num_style_feat self.final_linear = nn.Linear(channels['4']
unexpected key(s) in state_dict: "model".
12-15
"unexpected key(s) in state_dict: 'model'"是指在加载PyTorch模型参数时，在state_dict字典中发现了不匹配的键值对，其中“model”是一个未预期的键。这通常是由于加载模型时提供的state_dict与模型定义的结构不一致导致的。 要解决这个问题，我们可以采取以下步骤： 1. 确保模型定义和加载参数的方式保持一致。检查模型定义中是否使用了“model”作为模型类的属性或参数名称。如果是，我们应该在加载state_dict时使用相同的名称。 2. 检查模型定义是否包含了所有在state_dict中的键值对。可以通过打印模型定义中的state_dict.keys()和加载参数的state_dict.keys()进行比较。如果存在不匹配的键，则需要对模型定义进行相应的调整。 3. 检查模型定义与加载参数的版本是否一致。如果模型定义的结构发生了变化，例如添加或删除了某些层或参数，那么加载参数时会出现不匹配的键。在这种情况下，我们应该更新模型定义来与加载参数保持一致。 总结起来，当我们遇到“unexpected key(s) in state_dict: 'model'”的错误时，我们应该检查模型定义和加载参数的一致性，包括属性或参数名称、键值对的匹配以及模型版本的一致性，并相应地进行调整和更新。
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值