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

最新推荐文章于 2025-04-16 13:41:30 发布

zyd的小飞船

最新推荐文章于 2025-04-16 13:41:30 发布

阅读量569

点赞数 9

文章标签： python

本文链接：https://blog.youkuaiyun.com/weixin_44124168/article/details/135470926

版权

本文探讨了在处理deep_sort库时遇到的维度不匹配问题，重点提示读者对model.py文件第55行进行相应修改以确保代码兼容性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

zyd的小飞船

关注关注

9
点赞
踩
9

收藏

觉得还不错? 一键收藏
1
评论
分享

复制链接

分享到 QQ

分享到新浪微博

扫一扫
举报

举报

size mismatch for head.weight: copying a param with shape torch.Size([1000, 768]) from checkpoint

Green____hand的博客

08-01

832

我在github上的vit分类模型（）使用了自己的数据集，训练的时候发生报错。报错原因：训练模型时出现FC层不匹配，我的数据集是 3个类的，下载的预权重是ImageNet 1000个类的。

python报错: RuntimeError: Error(s) in loading state_dict for ResNet:size mismatch for conv1.weight: ..

syll_的博客

10-10

1404

比如我这里是加载了一个resnet的预训练模型，但是从网上下载的时候把3D的模型下成了2D的，可以看到参数少了一个维度，重新下载之后就解决啦。这些不管能不能看懂，强烈建议先换成cpu再跑一遍，显示出真实的错误原因。

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

RuntimeError: Error(s) in loading state_dict for GoogLeNet: size mismatch for fc.weight: copying a

qq_43384629的博客

04-09

1636

我注释的这一部分就是问题提示的部分，报错显示fc.weight和fc.bias，比对我的错误，看自己的是那个不匹配，修改成自己的就行。

RuntimeError: Error(s) in loading state_dict for ResNet: size mismatch for fc.weight: copyin

2501_90729325的博客

04-16

403

RuntimeError: Error(s) in loading state_dict for ResNet: size mismatch for fc.weight: copying a param with shape torch.Size([1000, 2048]) from checkpoint, the shape in current model is torch.Size([6, 2048]).

RuntimeError: Error(s) in loading state_dict for Net（让人心累的错误）

qq_39606051的博客

09-19

1277

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

RuntimeError: Error(s) in loading state_dict for Net:

weixin_39357271的博客

11-17

2588

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

2625

但是，这处的问题并不好改，如果也是使用yolov5_rotated代码训练，建议对照yolov5的训练代码，可以发现这个rotated框架在优化器部分的代码写的不完善，而且预训练判断部分的代码也有问题，即使成功训练起来，epoch数量也不是从0，1开始，所以主要还是框架考虑不够完善。查找了一些办法，基本上是关于 loading_state_dict()的用法，如何去处理加载的权重，开始直接定位到了ckpt[‘model’]的这几层的权重，强行将几层的权重维度匹配到model中，但是训练时仍然没有解决问题。

File "d:\train\cnn_test\voc2012_ti\voc2012_ti\F-rnn-voc2012.py", line 340, in train_model model.load_state_dict(torch.load('D:/train/cnn_test/data/best_model.pth')) ~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "D:\anaconda_envs\envs\cnn_test\Lib\site-packages\torch\nn\modules\module.py", line 2581, in load_state_dict raise RuntimeError( ...<3 lines>... ) RuntimeError: Error(s) in loading state_dict for CNNRNNModel: size mismatch for classifier.weight: copying a param with shape torch.Size([2, 256]) from checkpoint, the shape in current model is torch.Size([3, 256]). size mismatch for classifier.bias: copying a param with shape torch.Size([2]) from checkpoint, the shape in current model is torch.Size([3]).

最新发布

05-14

RuntimeError: Error(s) in loading state_dict for Net: size mismatch for classifier.4.weight: copying a param with shape torch.Size([751, 256]) from checkpoint, the shape in current model is torch....

Traceback (most recent call last): File "E:\BaiduNetdiskDownload\UNet_Demo\UNet_Demo\predict.py", line 17, in <module> unet = Unet() File "E:\BaiduNetdiskDownload\UNet_Demo\UNet_Demo\unet.py", line 79, in __init__ self.generate() File "E:\BaiduNetdiskDownload\UNet_Demo\UNet_Demo\unet.py", line 90, in generate self.net.load_state_dict(torch.load(self.model_path, map_location=device)) File "E:\Software\anaconda3\envs\JXW\lib\site-packages\torch\nn\modules\module.py", line 2189, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for Unet: size mismatch for final.weight: copying a param with shape torch.Size([2, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([10, 64, 1, 1]). size mismatch for final.bias: copying a param with shape torch.Size([2]) from checkpoint, the shape in current model is torch.Size([10]).

03-15

pretrained_dict = {k: v for k, v in pretrained_dict.items() if k in model_dict and v.size() == model_dict[k].size()} 然后更新当前模型的state_dict： model_dict.update(pretrained_dict) 最后加载回去： ...

有没有可能是传入的预训练模型不对？

02-28

RuntimeError: Error(s) in loading state_dict for FasterRCNN: size mismatch for roi_head.bbox_head.fc_cls.weight: copying a param with shape torch.Size([81, 1024]) from checkpoint, which has shape...

pytorch模型加载报错：Error(s) in loading state_dict for PGAT_Model:size mismatch for X

weixin_42264992的博客

04-26

1334

model.load_state_dict(torch.load(checkpoint_file) ) 报错：Error(s) in loading state_dict for PGAT_Model:size mismatch for X

Python运行报错_RuntimeError: Error(s) in loading state_dict for DataParallel:

weixin_51659315的博客

12-13

1481

Debug发现checkpoint['model_state_dict'']才是要删除的预训练模型对应的字典。

RuntimeError: Error(s) in loading state_dict for RNN:问题解决

qq_42282231的博客

01-20

4988

RuntimeError: Error(s) in loading state_dict for RNN:问题解决当我们在进行机器学习时，会出现加载训练好的模型参数出现错误的情况，以下便是我遇到的情况。从上图可见，在加载训练好的模型参数时出现了问题，报错意思很明显，即加载的参数名称和新实例化的模型参数名称不一样。首先，查看已训练好的模型参数，如下图所示。而新实例化的模型参数如下图所示。很明显，参数列表的名称不一样。我怀疑我没有保存进去，所以重新训练一遍模型，结果如下所示：果然。。。。

pytorch载入模型报错Error(s) in loading state_dict for Net:

pylyy的博客

07-04

432

可能是import net的问题，

解决解决Error(s) in loading state_dict for *** :Missing key(s) in state_dict:Unexpected key(s) in state_

Zhe_Lin的博客

10-07

1002

导致第一次错误运行时保存了之前的错误model，修改代码之后再次运行会加载原有错误的模型，导致两个模型参数不对应。解决办法就是删除你之前运行保存"pre_model_save_path"路径下的“.pt”文件，再次运行即可。

RuntimeError: Error(s) in loading state_dict for DistributedDataParallel报错解决

weixin_73838412的博客

12-22

445

但是我最后debug出来似乎都不是以上原因，而是因为我在一个网络（假设是net1）中嵌套了一个网络（net2），但是在train文件中，我进行了model.load_state_dict（）（这里想要加载的是net1的权重）这时候会可能会把这个权重加载到嵌套中的另一个网络（net2），而不是本身的网络（net1）中，所以我将这两个网络单独定义，然后各自创建实例，加载权重，就解决这个问题了。3.我修改了模型的结构，导致参数加载不匹配，需要在加载时指定一个参数，strict=false。

解决Error(s) in loading state_dict for *** :

小威的博客

07-26

3366

strict=)True的话就要求训练权重层数的键值与新构建的模型的权重层数名称完全符合，(strict=)False就没有这个要求了。使用ide（这里是vscode）的debug功能，查看一下newckpt中的内容，如下。可以看出内容和报错信息极其相似，然而字典中每个键前面都多了一个"module"。希望读者可以举一反三，通过debug，依据自己的报错信息来进行对应的修改。因此，抓住本质，只需要修改一些checkpoint中键的名字就好了。这篇与我类似，但他是少了个"module"，所以要补上。.....

RuntimeError: Error(s) in loading state_dict for

sgfsfgs的博客

03-06

2106

版本问题不对应，降低pyopengl 版本至3.1.4。

RuntimeError: Error(s) in loading state_dict for BNInception:size mismatch for

你若安好是晴天的博客

07-07

7387

Traceback (most recent call last): File "test_video.py", line 105, in <module> img_feature_dim=args.img_feature_dim, print_spec=False) File "/data2/zcq1/TRN-pytorch/models.py", line 43, in __init__ self._prepare_base_model(base_model) .