Unexpected key(s) in state_dict: “dense_block1.denselayer1.norm.1

最新推荐文章于 2024-06-12 23:31:22 发布

AI算法网奇

最新推荐文章于 2024-06-12 23:31:22 发布

阅读量2.3k

点赞数

分类专栏： pytorch知识宝典

本文链接：https://blog.csdn.net/jacke121/article/details/112165494

版权

pytorch知识宝典专栏收录该内容

499 篇文章 226 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

Unexpected key(s) in state_dict: "dense_block1.denselayer1.norm.1

from torchvision.models import densenet121
from collections import OrderedDict

model = densenet121(pretrained=False)

state_dict =torch.load(model_weight_path)
# 初始化一个空 dict
new_state_dict = OrderedDict()
# 修改 key
for k, v in state_dict.items():
    if 'denseblock' in k:
        param = k.split(".")
        k = ".".join(param[:-3] + [param[-3]+param[-2]] + [param[-1]])
    new_state_dict[k] = v
    model.load_state_dict(new_state_dict)

我的解决方法：

# 初始化一个空 dict
new_state_dict = OrderedDict()
# 修改 key
for k, v in state_dict.items():

了解本专栏

订阅专栏解锁全文

超级会员免费看

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

AI算法网奇

关注关注

0
点赞
踩
1

收藏

觉得还不错? 一键收藏
打赏
2
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

专栏目录

订阅专栏

LLM：ChatGLM-6B模型文件modeling_chatglm.py讲解

u013250861的博客

08-09

463

【代码】LLM：ChatGLM-6B模型文件modeling_chatglm.py讲解。

Pysot demo.py载入模型报错 “Missing key(s) in state_dict”的解决方案

qq_40313336的博客

04-20

2173

如何在Pysot的demo.py中加载自己训练后的模型

2 条评论您还未登录，请先登录后发表或查看评论

加载pytorch预训练模型densenet121出现Missing key(s) in state_dict Unexpected key(s) in state_dict

weixin_41693877的博客

08-04

2034

在手动加载densenet121权重时，报了一下的错误：

pytorch错误解决：Missing key(s) in state_dict: Unexpected key(s) in state_dict:

热门推荐

u013925378的博客

03-09

2万+

在进行模型测试时报错： Missing key(s) in state_dict: xxxxxxxxxx Unexpected key(s) in state_dict:xxxxxxxxxx 报错原因：在模型训练时有加上：【可以加速训练速度】 model = nn.DataParallel(model) #cudnn.benchmark = True 但是在模型测试推断时，在...

pytorch load_state_dict模型，报错“Missing key(s) in state_dict、Unexpected key(s) in state_dict”

acrith

10-22

2155

通过pytorch 平台训练一个checkpoint（xxx.pth），再infer或者finetune时，利用load_state_dict加载xxx.pth，如果报一下错误：那么，需要检查下，在训练时，是否加入了一下语句： net = torch.nn.DataParallel(net) 如果，加入了以上语句，那么，在infer或者finetune时，同样需要加入以上语句，即可解决报错。 ...

pytorch导入训练好的模型时报错Missing key(s) in state_dict,Unexpected key(s) in state_dict

qq_35037684的博客

10-09

5018

完整报错信息如下： File "F:/pytorch/anti-spoofing_depth/living_test.py", line 53, in run_test network.load_state_dict(torch.load(initial_checkpoint, map_location=lambda storage, loc: storage)) File "D:\anaconda\lib\site-packages\torch\nn\modules\module.py", l

LLM-微调：Peft库--get_peft_model()--＞在llm基座模型的基础上注入Lora模块（加载流程）【注入的Lora模块的初始参数是随机初始化的】

u013250861的博客

10-22

1259

【代码】LLM-微调：Peft库--get_peft_model()函数【在llm基座模型的基础上加载Lora模块（Lora模块的初始参数是随机初始化的）】

【YOLOv8改进 - Backbone主干】FasterNet：基于PConv（部分卷积）的神经网络，提升精度与速度，降低参数量。

06-12

426

【YOLOv8改进 - Backbone主干】FasterNet：基于PConv（部分卷积）的神经网络，提升精度与速度，降低参数量。许多研究工作集中在减少浮点运算次数（FLOPs）。我们观察到，尽管减少FLOPs确实带来了一定的性能提升，但并不一定会导致相同程度的延迟减少。这主要是由于浮点运算每秒（FLOPS）效率低下导致的。为了实现更快的网络，我们重新审视了流行的算子，并发现这种低FLOPS主要是由于算子的频繁内存访问，特别是深度卷积。

bert源码(pytorch)超详细解读！！！

ZJF的博客

05-05

3273

modeling.py 此文章是对transformers的bert源码的解读 # coding=utf-8 from __future__ import absolute_import, division, print_function, unicode_literals import copy import json import logging import math import os import shutil import tarfile import tempfile import sys

pytorch Missing key(s) in state_dict

xiakj的专栏

02-11

1175

加载模型的时候，加上一个关键字 strict model.load_state_dict(pretrained_dict, strict=False) # 多GPU训练，单GPU推理 strict=False

RuntimeError: Error(s) in loading state_dict for (Missing key(s) in state_dict: “firstconv.weight)

深度菜鸡-达闻西的博客

07-20

736

if use_gpu: model = torch.nn.DataParallel(model, device_ids=[0, 1])#####加上这句 model = model.cuda() model = load_network(model)########再加载参数已解决

pytorch加载模型报错RuntimeError: Error(s) in loading state_dict for SSD:Missing key(s) in state_dict:...

qq_40502460的博客

03-26

3474

报错信息： RuntimeError: Error(s) in loading state_dict for SSD: Missing key(s) in state_dict: “backbone.vgg.0.weight”, “backbone.vgg.0.bias”, “backbone.vgg.2.weight”, “backbone.vgg.2.bias”, “backbone.vgg.5.weight”, “backbone.vgg.5.bias”, “backbone.vgg.7.weight

Densenet121预训练权重加载不匹配问题

m0_57201352的博客

11-17

1058

权重改名

RuntimeError: Error(s) in loading state_dict for (Missing key(s) in state_dict: "firstconv.weight)

qq_20373723的博客

01-09

3627

错误记录太长了，解决在最下面 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"...

Missing key(s) in state_dict

l2181265的博客

10-11

4015

这种错误多半是模型加载错误导致。排查测试代码的模型，与加载的模型是否是同一个。错误代码： Missing key(s) in state_dict: "input_block.conv1.conv.weight",... Unexpected key(s) in state_dict: "module.input_block.conv1.conv.weight",... 错误分析：区别在于一个加了module，一个没加。解决办法：训练模型时，在服务器使用了多GPUtorch.nn.DataPara

同一个模型，使用nn.DataParallel前后，pytorch加载参数文件时报错：missing key(s) in state_dict

A_Student10000的博客

11-23

552

同一个模型，pytorch加载参数文件时报错：missing key(s) in state_dict

pytorch读取模型失败RuntimeError: Error(s) in loading state_dict for ResNet: Missing key(s) in state_dict

路过人间，一叶知秋

03-21

3525

错误 RuntimeError: Error(s) in loading state_dict for ResNet: Missing key(s) in state_dict: Missing key(s) in state_dict: “conv1.weight”, “bn1.weight”, Unexpected key(s) in state_dict: “module.conv1.weight”, “module.bn1.weight” 原因在训练模型的时候使用了torch.nn.DataP

RuntimeError: Error(s) in loading state_dict for ResNet: Missing key(s) in state_dict

weixin_39386156的博客

11-29

1469

加载预训练模型

Missing key(s) in state_dict: "conv1.weight" Unexpected key(s) in state_dict: "model.conv1.weight",

06-09

例如，如果预训练模型中的key为"model.conv1.weight"，而当前模型中的key为"conv1.weight"，可以通过以下代码进行加载： ```python pretrained_dict = torch.load('pretrained_model.pth') model_dict = model....