size mismatch for xx.weight错误的解决方法

问题重现：

RuntimeError: Error(s) in loading state_dict for xxxNet:size mismatch for bn1.weight: copying a param with shape torch.Size([3]) from checkpoint, the shape in current model is torch.Size([512]).size mismatch for bn1.bias: copying a param with shape torch.Size([3]) from checkpoint, the shape in current model is torch.Size([512]).size mismatch for bn1.running_mean: copying a param with shape torch.Size([3]) from checkpoint, the shape in current model is torch.Size([512]).size mismatch for bn1.running_var: copying a param with shape torch.Size([3]) from checkpoint, the shape in current model is torch.Size([512]).

问题分析：

在使用load_state_dict(model_dict, False)时会报错，原因是你现有的模型和你权重文件的保存的模型结构不一样，但是我现有的模型结构和原来是同一个，为什么还会报错呢？
原因是：你现有的模型内有一些已定义，但是没有使用的层，比如在我上面的报错中，bn1层在我们模型的__init__里面定义了，但是没有在forward中使用，但是load_state_dict()仍然会把权重文件.pth里面的bn层赋值给bn1层，导致层的对应关系错误，所以要把bn1去掉就可以了。

size mismatch for xx.weight错误的解决方法相关推荐

【Python】解决CNN中训练权重参数不匹配size mismatch for fc.weight，size mismatch for fc.bias
目录 1.问题描述 2.问题原因 3.问题解决 3.1思路1--忽视最后一层权重额外说明:假如载入权重不写strict=False, 直接是model.load_state_dict(pre_wei ...
size mismatch for fc.weight: copying a param with shape torch.Size([1000, 2048]) from checkpoint, th
问题描述我想在我自己的项目更换其他的模型,下载的预训练模型出现了FC层不匹配的问题,找了好多人都写了这个点,今天总结一下: 首先我们遇到的问题如下: 他的意思是resnet50的fc层是1000分类 ...
centos linux 系统日常管理4 scp,rsync,md5sum,sha1sum,strace ,find Rsync 常见错误及解决方法第十七节课...
centos linux 系统日常管理4 scp,rsync,md5sum,sha1sum,strace ,find Rsync 常见错误及解决方法第十七节课 rsync可以增量同步,scp不行 ...
Android 导入项目时出现错误的解决方法（红色感叹号）
如果发现导入工程的时候,出现"No projects are found to import" 的提示,首先查看项目目录中是否有隐藏文件.project,还有目录结构也还要有一个隐 ...
python运行py文件参数错误_Pycharm运行加载文本出现错误的解决方法
Pycharm打开大文件的时候出现这个错误,根据提示以及百度知道,是因为IDEA对能关联的文件大小做了限制,主要是为了保护内存,默认值为2500kb.然后根据百度上的教程修改修改bin文件中的ide ...
【安全】Kerberos相关问题进行故障排除| 常见错误和解决方法
1.概述转载为了学习:Kerberos相关问题进行故障排除| 常见错误和解决方法 2.总结可以用来帮助诊断Kerberos相关问题的原因并实施解决方案的指南. 3. 症状单击症状链接转到相应的 ...
mysql 1058_MySQL数据库之mysql启动服务报1058错误的解决方法
本文主要向大家介绍了MySQL数据库之mysql启动服务报1058错误的解决方法 ,通过具体的内容向大家展现,希望对大家学习MySQL数据库有所帮助. 今天研究一个框架,数据库使用mysql,启动my ...
ansys 服务器无响应,ANSYS出现的各种错误及解决方法
原标题:ANSYS出现的各种错误及解决方法 1.The value of UY at node 1195 is 449810067.It is greater than the current lim ...
mysql 1058_MySQL数据库之mysql启动服务报1058错误的解决方法
本文主要向大家介绍了MySQL数据库之mysql启动服务报1058错误的解决方法 ,通过具体的内容向大家展现,希望对大家学习MySQL数据库有所帮助. 今天研究一个框架,数据库使用mysql,启动my ...

size mismatch for xx.weight错误的解决方法

问题重现：

问题分析：

size mismatch for xx.weight错误的解决方法相关推荐

最新文章

热门文章