KeyError: ‘layer1.1.bn1.bias‘

在跑AICITY2020_DMT_HST的代码时遇到的问题，记录一下

参考文章

问题描述

加载resnet101模型的时候，参数多了

解决方法

将不用的参数过滤掉

#原代码
# def load_param(self, model_path):#     param_dict = torch.load(model_path)#     if 'state_dict' in param_dict:#         param_dict = param_dict['state_dict']#     for i in param_dict:#         if 'fc' in i:#             continue#         embed()#         self.state_dict()[i.replace('module.','')].copy_(param_dict[i])#改的代码   def load_param(self, model_path):param_dict = torch.load(model_path)param_dict = {k: v for k, v in param_dict.items() if 'bn1.running_mean' not in k}if 'state_dict' in param_dict:param_dict = param_dict['state_dict']for i in param_dict:# embed()if 'layer1.0.bn1.running_mean' in i:continueif 'fc' in i:continueif 'bn1.running_var' in i:continueif 'bn1.weight' in i:continueif 'bn1.bias' in i:continueif 'layer1.1.bn1.running_var' in i:continueif 'layer1.1.bn1.weight'  in i:continueself.state_dict()[i.replace('module.', '')].copy_(param_dict[i])

KeyError: ‘layer1.1.bn1.bias‘相关推荐

Unexpected key(s) in state_dict: “module.conv1.weight“, “module.bn1.weight“, “module.bn1.bias“,
由于服务器老是断电所以想加载已经训练好的上一个epoch的模型,但是在加载时遇到了这个问题这是由于保存模型字典时每一个模块的key都自动加上了'module'.所以在加载模型参数继续训练时就会与模 ...
pytorch加载预训练模型遇到的问题:KeyError: ‘bn1.num_batches_tracked‘
问题最近在使用pytorch1.0加载resnet预训练模型时,遇到的一个问题,在此记录一下. KeyError: 'layer1.0.bn1.num_batches_tracked' 其实 ...
初始化模型参数 python_pytorch 网络参数 weight bias 初始化详解_python_脚本之家
权重初始化对于训练神经网络至关重要,好的初始化权重可以有效的避免梯度消失等问题的发生. 在pytorch的使用过程中有几种权重初始化的方法供大家参考. 注意:第一种方法不推荐.尽量使用后两种方法. # ...
载入模型部分权重的方法
如果我们对模型进行微调,比如改变模型的最后一个全连接层输出节点个数,此时我们再载入原来模型的预训练权重,就会报错.因为模型的结构已经发生了变化,所以它所对应的权重(通常以字典中键值对的形式存储)存储结 ...
Pytorch-模型参数：named_parameters()、parameters()、state_dict()区别
torch中存在3个功能极其类似的方法,它们分别是model.parameters().model.named_parameters().model.state_dict(),下面就具体来说说这三个函 ...
【论文复现】使用PaddleDetection复现OrientedRepPoints的复现笔记
1 复现流程复现流程表: 翻译原始论文: 学习PaddleDetection配置参数对齐Dataloader: 2 MMRotate代码 2.1 配置mmrotate环境官方安装文档:INSTA ...
pytorch获取网络的层数和每层的名字
#创建自己的网络 import models model = models.__dict__["resnet50"](pretrained=True)for index ,(nam ...
pytorch中的model.named_parameters()与model.parameters()
参考链接:https://www.cnblogs.com/yqpy/p/12585331.html model.named_parameters() 迭代打印model.named_parameter ...
pytorch加载模型报错Unexpected key(s) in state_dict: module.conv1.weight, module.bn1
文章目录背景报错原因解决背景 Pytorch在加载模型参数的时候,有两种情况可能出现这种问题: 自己写的网络结构,例如: 代码 import models arch = 'resnet50' ...

KeyError: ‘layer1.1.bn1.bias‘

问题描述

解决方法

KeyError: ‘layer1.1.bn1.bias‘相关推荐

最新文章

热门文章