Unexpected key(s) in state_dict: “dense

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

from torchvision.models import densenet121
from collections import OrderedDictmodel = 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] = vmodel.load_state_dict(new_state_dict)

我的解决方法：

# 初始化一个空 dict
new_state_dict = OrderedDict()
# 修改 key
for k, v in state_dict.items():k=k.replace('module.', '')if 'dense_block' in k:if "norm" in k or "conv.1" in k or "conv.2" in k:param = k.split(".")k = ".".join(param[:-3] + [param[-3]+param[-2]] + [param[-1]])new_state_dict[k] = velse:new_state_dict[k] = v

Unexpected key(s) in state_dict: “dense_block1.denselayer1.norm.1相关推荐

PyTorch加载模型model.load_state_dict()问题，Unexpected key(s) in state_dict: “module.features..，Expected .
希望将训练好的模型加载到新的网络上.如上面题目所描述的,PyTorch在加载之前保存的模型参数的时候,遇到了问题. Unexpected key(s) in state_dict: "mod ...
Unexpected key(s) in state_dict: module.backbone.bn1.num_batches_tracked
pytorch预测的时候报异常了: {RuntimeError}Error(s) in loading state_dict for DataParallel: Unexpected key( ...
Unexpected key(s) in state_dict: “module.conv1.weight“, “module.bn1.weight“, “module.bn1.bias“,
由于服务器老是断电所以想加载已经训练好的上一个epoch的模型,但是在加载时遇到了这个问题这是由于保存模型字典时每一个模块的key都自动加上了'module'.所以在加载模型参数继续训练时就会与模 ...
pytorch加载模型报错Unexpected key(s) in state_dict: module.conv1.weight, module.bn1
文章目录背景报错原因解决背景 Pytorch在加载模型参数的时候,有两种情况可能出现这种问题: 自己写的网络结构,例如: 代码 import models arch = 'resnet50' ...
Huggingface填坑：Unexpected key(s) in state_dict: “roberta.pooler.dense.weight“....
在加载已经训练好的模型时,出现问题如下: RuntimeError: Error(s) in loading state_dict for RobertaForSequenceClassificati ...
Missing key(s) in state_dict | unexpected_keys | 模型加载 [torch.load] 报错【解决方案】
文章目录 Missing key(s) in state_dict 报错分析该报错快速直接解决方法额外思路 [Unexpected key(s) in state_dict]- 解决思路 Miss ...
Missing key(s) in state_dict
Missing key(s) in state_dict 解决方法: try:from collections import OrderedDictnew_state_dict = OrderedDi ...
命名实体识别NER遗留问题----模型构建
深度学习模型预测实质:训练保存的模型里面参数整个只有一套参数不仅保存了训练数据全部的正确信息,而且同字多义的情况下通过其同行的词来判断,虽然参数都是一套但是因为输入的值不同导致计算的结果不同导致 ...
Pytorch模型迁移和迁移学习,导入部分模型参数
Pytorch模型迁移和迁移学习目录 Pytorch模型迁移和迁移学习 1. 利用resnet18做迁移学习 2. 修改网络名称并迁移学习 3.去除原模型的某些模块 1. 利用resnet18做迁移 ...

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

Unexpected key(s) in state_dict: “dense_block1.denselayer1.norm.1相关推荐

最新文章

热门文章