RuntimeError: Error(s) in loading state_dict for Net: size mismatch for classifier.4.weight: copying

于 2024-01-09 08:40:53 发布
阅读量470 收藏 9
点赞数 9
文章标签： python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_44124168/article/details/135470926
版权
本文探讨了在处理deep_sort库时遇到的维度不匹配问题，重点提示读者对model.py文件第55行进行相应修改以确保代码兼容性。
摘要由CSDN通过智能技术生成

分类器维度不匹配，修改deep_sort/deep/model.py 55行左右：

RuntimeError: Error(s) in loading state_dict for ResNet: Unexpected key(s) in state_dict: “bn1.num
CSDN 精品推荐
01-09 1188
在这个特定的情况下，错误信息中提到了多个“num_batches_tracked”键。这类键在某些 PyTorch 版本中存在，而在其他版本中可能不存在。方法会加载所有匹配的键，而忽略任何不匹配的键，从而避免了这个错误。但是，这种方法应谨慎使用，因为它可能掩盖模型结构和状态字典之间的不一致问题。这样做会使得 PyTorch 忽略不匹配的键，只加载那些与模型结构相匹配的键。这通常发生在模型的结构和要加载的状态字典不完全匹配时。解决这个问题的一个常见方法是在加载状态字典之前，明确地忽略不匹配的键。
RuntimeError: Error(s) in loading state_dict for SSD
wlw8023diya的博客
07-30 1744
error: (pytorch) ubuntu-gpu@UbuntuGPU:~/wangmiao/ssd-pytorch-master$ python get_dr_txt.py Loading weights into state dict... Traceback (most recent call last): File "get_dr_txt.py", line 131, in <module> ssd = mAP_SSD() File "/home/ubuntu-
RuntimeError: Error(s) in loading state_dict for & size mismatch for
wangmengmeng99的博客
10-24 4365
RuntimeError: Error(s) in loading state_dict for & size mismatch for
RuntimeError: Error(s) in loading state_dict for Net（让人心累的错误）
qq_39606051的博客
09-19 1202
RuntimeError: Error(s) in loading state_dict for Net: size mismatch for classifier.4.weight
RuntimeError: Error(s) in loading state_dict for ft_net:
qq_36414085的博客
11-14 2579
一、我出现这个错误的原因是： 我在加载之前的模型，想进行再次训练，但是model.load_state_dict(torch.load(“要加载的路径”))这一步就开始报上面的错误。 原因是我的模型之前加了 PCB，但是现在重新训练的代码要求不用PCB，导致之前的model不适用现在的代码环境出错。 解决：要么将现在的代码PCB那句话改成用PCB， 要么将之前的model重新训练不加PCB的，适合...
RuntimeError: Error(s) in loading state_dict for Net:
weixin_39357271的博客
11-17 2521
size mismatch for classifier.4.weight: copying a param with shape torch.Size([7, 256]) from checkpoint, the shape in current model is torch.Size([751, 256]). size mismatch for classifier.4.bias: copying a param with shape torch.Size([7]) from checkpoint, t
RuntimeError: Error(s) in loading state_dict for Model: size mismatch for model.24.m.0.weight....
qq_44442727的博客
11-07 2326
但是，这处的问题并不好改，如果也是使用yolov5_rotated代码训练，建议对照yolov5的训练代码，可以发现这个rotated框架在优化器部分的代码写的不完善，而且预训练判断部分的代码也有问题，即使成功训练起来，epoch数量也不是从0，1开始，所以主要还是框架考虑不够完善。查找了一些办法，基本上是关于 loading_state_dict()的用法，如何去处理加载的权重，开始直接定位到了ckpt[‘model’]的这几层的权重，强行将几层的权重维度匹配到model中，但是训练时仍然没有解决问题。
RuntimeError: Error(s) in loading state_dict for ResNet50:size mismatch for classifier.weight
qq_41876456的博客
03-29 2336
问题思路 加载的模型的分类器和现在训练的分类不一致 解决问题 把分类器那里改成一样的就好了 # model = models.init_model(name=args.arch, num_classes=dataset.num_train_pids, loss={'xent', 'htri'}) model = models.init_model(name=args.arch, num_classes=767, loss={'xent', 'htri'}) 这种改法就比较简单粗暴了，可能会有更巧妙
参数不匹配问题：RuntimeError: Error(s) in loading state_dict for YoloBody: size mismatch for last_layer0.6。。
weixin_44239195的博客
06-29 9604
问题描述 题主主要是在pytorch上跑yolov3 时出现如下问题，经过一次次的bug，各种百度，各种调试之后，写下此博客，希望能有所帮助。才疏学浅，欢迎批评指正 出现的错误主要如下 RuntimeError: Error(s) in loading state_dict for YoloBody: size mismatch for last_layer0.6.weight: copying a param with shape torch.Size(【255, 1024, 1, 1】) from ch
torch RuntimeError: Error(s) in loading state_dict for CRNN:
hzhj的博客
01-13 3479
加载训练的模型报如下错误。 >>> model.load_state_dict(torch.load(model_path)) Traceback (most recent call last): File "<stdin>", line 1, in <module> File "/usr/local/lib/python2.7/dist-pa...
Python运行报错_RuntimeError: Error(s) in loading state_dict for DataParallel:
weixin_51659315的博客
12-13 1376
Debug发现checkpoint['model_state_dict'']才是要删除的预训练模型对应的字典。
【解决报错】RuntimeError: Error(s) in loading state_dict for Net:size mismatch for classifier.
weixin_47214888的博客
04-11 1829
更改代码位置：deep_sort/deep/model.py 49行位置。报错原因是分类器维度不匹配，将10改成751即可。更改完成后可正常运行。
pytorch载入模型报错Error(s) in loading state_dict for Net:
pylyy的博客
07-04 393
可能是import net的问题，
RuntimeError: Error(s) in loading state_dict for RNN:问题解决
qq_42282231的博客
01-20 4867
RuntimeError: Error(s) in loading state_dict for RNN:问题解决 当我们在进行机器学习时，会出现加载训练好的模型参数出现错误的情况，以下便是我遇到的情况。 从上图可见，在加载训练好的模型参数时出现了问题，报错意思很明显，即加载的参数名称和新实例化的模型参数名称不一样。 首先，查看已训练好的模型参数，如下图所示。 而新实例化的模型参数如下图所示。 很明显，参数列表的名称不一样。 我怀疑我没有保存进去，所以重新训练一遍模型，结果如下所示： 果然。。。。
解决RuntimeError：Error（s）in loading state_dict for YoloBody
热门推荐
xhxzdxm的博客
04-05 1万+
解决RuntimeError：Error（s）in loading state_dict for YoloBody问题描述分析解决方法 问题描述 本人运行的是官网下载的基于pytorch实现的YOLOV3模型。在尝试利用训练好的权值文件去进行预测时产生一下问题： RuntimeError: Error(s) in loading state_dict for YoloBody: size mismatch for last_layer0.6.weight: copying a param with sha
【解决bug】RuntimeError: Error(s) in loading state_dict for Net: size mismatch for classifier
weixin_44459446的博客
01-18 2814
在feature_extractor.py 第12行 改成。使用deepsort检测自己的数据集时遇到。
Error(s) in loading state_dict for ResNet 问题解决
呆呆的猫的博客
07-02 9941
当使用如下命令加载模型的时候，会报错，没有一些 keys： model.load_state_dict(model_zoo.load_url(model_urls['resnet101'])) RuntimeError: Error(s) in loading state_dict for ResNet: Missing key(s) in state_dict: "C1_down_channel.weight", "C1_down_channel.bias", "C2_down_channel.wei
RuntimeError: Error(s) in loading state_dict for BertClassifier 模型不匹配
fghjbjhgb的博客
08-15 4305
RuntimeError: Error(s) in loading state_dict for BertClassifier: size mismatch for lstm.weight_ih_l0: copying a param with shape torch.Size([400, 768]) from checkpoint, the shape in current model is torch.Size([360, 768]). size mismatch for lstm.weight_h
RuntimeError: Error(s) in loading state_dict for UNet: Unexpected key(s) in state_dict:
最新发布
12-14
当出现"RuntimeError: Error(s) in loading state_dict for UNet: Unexpected key(s) in state_dict"错误时，这意味着在加载模型参数时，state_dict中的键与模型中的键不匹配。这可能是由于模型结构的更改或使用不同的模型架构导致的。为了解决这个问题，可以使用以下方法： 1.检查模型结构是否与保存的模型参数相同。如果模型结构已更改，则需要相应地更改保存的模型参数。 2.如果模型结构相同，则需要检查state_dict中的键和模型中的键是否匹配。可以使用以下代码检查它们： ```python model = UNet() state_dict = torch.load(PATH) for k in state_dict.keys(): if k not in model.state_dict().keys(): print('Unexpected key in state_dict:', k) ``` 3.如果发现不匹配的键，则需要将state_dict中的键重命名为模型中的键。可以使用以下代码实现： ```python model = UNet() state_dict = torch.load(PATH) new_state_dict = {} for k, v in state_dict.items(): name = k if k.startswith('module.'): name = k[7:] # 去掉'module.'前缀 new_state_dict[name] = v model.load_state_dict(new_state_dict) ``` 以上是解决"RuntimeError: Error(s) in loading state_dict for UNet: Unexpected key(s) in state_dict"错误的方法。
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值