KeyError: ‘layer1.1.bn1.bias‘

最新推荐文章于 2023-04-15 11:11:47 发布

Coder_L2

最新推荐文章于 2023-04-15 11:11:47 发布

阅读量1.3k

点赞数

分类专栏：问题解决文章标签：自动驾驶人工智能机器学习

本文链接：https://blog.csdn.net/qq_41876456/article/details/121632488

版权

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

参考文章

问题描述

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

解决方法

将不用的参数过滤掉

#原代码
# def load_param(self, model_path):
    #     param_dict = torch.load(model_path)
    #     if 'state_dict' in param_dict:
    #         param_dict

最低0.47元/天解锁文章

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

Coder_L2

关注关注

0
点赞
踩
0

收藏

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

分享到 QQ

分享到新浪微博

扫一扫

专栏目录

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

pytorch加载模型时报错

qq_39076073的博客

06-03

338

pyorch加载模型是出现错误： model.load_state_dict(checkpoint['state_dict']) 错误： RuntimeError: Error(s) in loading state_dict for SiameseNet: Missing key(s) in state_dict: "CNN.pretrained.conv1.weight", "CNN.pretrained.bn1.weight", "CNN.pretrained.bn1.bias", "CNN.pr

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

Unexpected key(s) in state_dict: “module.conv1.weight“, “module.bn1.weight“, “module.bn1.bias“,

qq_44169314的博客

03-20

3674

由于服务器老是断电所以想加载已经训练好的上一个epoch的模型，但是在加载时遇到了这个问题这是由于保存模型字典时每一个模块的key都自动加上了‘module’。所以在加载模型参数继续训练时就会与模型对不上号。 RuntimeError: Error(s) in loading state_dict for ResNet: Unexpected key(s) in state_dict: "module.conv1.weight", "module.bn1.weight", "module.bn1.bi

多GPU训练保存的参数，单GPU的环境，KeyError: ‘base.conv1.weight‘

qq_42820467的博客

06-08

1421

def load_param(self, model_path): param_dict = torch.load(model_path) # ============================================================================= # for i in param_dict: # if 'fc' in i: # continue #

可视化pytorch 模型中不同BN层的的running mean曲线

Nick Blog

06-01

1256

加载模型字典逐一判断每一层，如果该层是bn 的 running mean，就取出参数并取平均作为该层的代表对保存的每个BN层的数值进行曲线可视化 from functools import partial import pickle import torch import matplotlib.pyplot as plt pth_path = 'checkpoint.pth' pickle.load = partial(pickle.load, encoding="latin1") pickl.

【KeyError: ‘unexpected key “module.*****“】【KeyError: ‘unexpected key “net.“】

github_38575699的博客

04-15

217

下载的预训练模型显示key不对，

Pytorch 按层调整学习率

weixin_43118280的博客

04-08

1380

Pytorch 按层调整学习率第一步：先获取每一层的一个layer_scale layer_scales = list(layer_decay ** (num_layers - i) for i in range(num_layers + 1)) 第二步：把

网络结构——deeplab_v2_vgg

guohui0412的博客

09-21

457

** deeplab_v2_vgg ** 下面展示一些内联代码片。 import math import torchvision import torch import torch.nn as nn import torch.nn.functional as F class DeepLabASPPVGG(nn.Module): """Adapted from official implementation: http://liangchiehchen.com/projects/De

pytorch wrong when predicting using trained model

小膜仙的博客

05-02

305

RuntimeError: Error(s) in loading state_dict for ResNet: Missing key(s) in state_dict: "conv1.weight", "bn1.running_var", "bn1.running_mean", "bn1.bias", "bn1.weight", "layer1.0.conv1.weight", "layer...

解决pytorch多GPU训练保存的模型,在单GPU环境下加载出错问题

01-21

背景在公司用多卡训练模型，得到权值文件后保存，然后回到实验室，没有多卡的环境，用单卡训练，加载模型时出错，因为单卡机器上，没有使用DataParallel来加载模型，所以会出现加载错误。原因 DataParallel包装的模型在保存时，权值参数前面会带有module字符，然而自己在单卡环境下，没有用DataParallel包装的模型权值参数不带module。本质上保存的权值文件是一个有序字典。解决方法 1.在单卡环境下，用DataParallel包装模型。 2.自己重写Load函数，灵活。 from collections import OrderedDict def myOwnLoa

replace函数和*的使用

weixin_41998239的博客

07-31

539

从输出中可以看到，第一个函数中，*b的位置可以传入任意多没有关键字的参数，*b会将这些传入参数转化成一个元组，下面的调用。count可选参数，指定count后，只有str中前count个旧字符串old被替换。表示该位置接受任意多个非关键字（non-keyword）参数，在函数中将其转化为。"""a是一个普通传入参数，*b是一个非关键字星号参数"""new用于替换str中的新字符串；oldstr中要被替换的旧字符串；该函数执行完毕后，将生成替换后的字符串。......

pytorch 换版本_Pytorch 模型版本切换

weixin_29313547的博客

01-12

3227

0.3.1转到0.4.1或更高版本直接使用代码导入时常碰到 ‘BatchNorm2d’ object has no attribute ‘track_running_stats’的报错信息，这是由于0.3.1中的BN操作中没有配置track_running_stats参数,0.3.1中BatchNorm的定义如下class torch.nn.BatchNorm2d(num_features, ep...

错误：ValueError: Variable layer1-conv1/weight already exists

热门推荐

xiaohuihui1994的博客

06-27

1万+

当在Spyder下执行LeNet5.py文件： -- coding: utf-8 - import os import numpy as np import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data 加载mnist_inference.py中定义的常量和前向传播的函数 imp...

pytorch加载预训练模型遇到的问题:KeyError: ‘bn1.num_batches_tracked‘

asuradayuci的博客

07-16

9598

问题最近在使用pytorch1.0加载resnet预训练模型时,遇到的一个问题，在此记录一下。 KeyError: 'layer1.0.bn1.num_batches_tracked’ 其实是使用的版本的问题，pytorch0.4.1之后在BN层加入了track_running_stats这个参数, 这个参数的作用如下: 训练时用来统计训练时的forward过的min-batch数目,每经过一个min-batch, track_running_stats+=1 如果没有指定momentum

关于Pytorch的parameters()里面，BN layer没有running_mean和running_var

Do not mess it up

02-13

7743

在Pytorch的parameters()里面，BN layer只有2个可学习的参数alpha和belta，没有running_mean和running_var这2个统计量但是net.state_dict()是有running_mean和running_var的因为running_mean和running_var不是可以学习的变量，只是训练过程对很多batch的数据统计 ref：h...

（９）训练时报错KeyError: ‘conv1.1.0.weight‘

weixin_40227656的博客

07-17

8976

将B_mobilenet_yolov4_lite的PANet改为原来的3x3卷积，命名为L_mobilenet-yolov4_VOC，即将B_yolov4的主干网络换为mobilenet，训练时报错： Traceback (most recent call last): File "/home/lidan/PycharmProjects/yolov4/L_mobilenet-yolov4_VOC_0717/train.py", line 221, in <module> pret

解决报错： raise KeyError('missing keys in state_dict: "{}"'.format(missing))

muli

08-13

2671

报错如下： File “/home/muli/CAM-Experiment/attention_branch_network-pytorch/resnet.py”, line 309, in resnet101 model.load_state_dict(pretrained_dict) File “/home/muli/anaconda3/lib/python3.5/site-packages/...

RuntimeError: Error(s) in loading state_dict for YoloBody: Missing key(s) in state_dict: "backbone.conv1.weight", "backbone.bn1.weight", "backbone.bn1.bias", "backbone.bn1.running_mean", "backbone.bn1.running_var",