RuntimeError: Error(s) in loading state_dict for SSD: Missing key(s) in解决办法

最新推荐文章于 2024-05-05 11:30:08 发布
最新推荐文章于 2024-05-05 11:30:08 发布
阅读量8.5k 收藏 13
点赞数 22
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/anne332/article/details/109325140
版权

模型验证的时候运行出现RuntimeError: Error(s) in loading state_dict for SSD:Missing key(s) in state_dict: 解决办法|

将

net.load_state_dict(torch.load(args.trained_model))

改为

net.load_state_dict(torch.load(args.trained_model), False)

即可

主要是记录一下自己平时遇到的问题，和大家分享一下
如有侵犯，请联系我
点个赞支持一下吧

Anne332
关注
  • 22
    点赞
  • 踩
  • 13
    收藏
    觉得还不错? 一键收藏
  • 14
    评论
RuntimeError: Error(s) in loading state_dict for SSD
wlw8023diya的博客
07-30 1673
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-
Python RuntimeError: thread.__init__() not called解决方法
09-22
然而，在使用`threading.Thread`进行多线程编程时，可能会遇到`RuntimeError: thread.__init__() not called`这样的错误。这个错误通常意味着你没有正确地初始化`threading.Thread`的实例。 在标题和描述中提到的...
RuntimeError: Error(s) in loading state_dict for DistributedDataParallel: Missing key(s) in state_di
weixin_57948968的博客
05-05 413
这个错误通常是因为键值对不匹配的问题，也就是说在训练过程中，你保存的checkpoints文件中的键值对对应不上，这个时候你需要把原来的训练的文件删除，然后再重新进行训练即可。
（已解决）RuntimeError: Error(s) in loading state_dict for Generator: Missing key(s) in state_dict: “hpf
qq_45937473的博客
03-13 1573
这个错误表明在加载模型的状态字典时出现了问题，因为状态字典中缺少了一个名为"hpf.filter"的键。这通常是由于尝试加载的模型与代码中期望的模型结构不匹配所导致的。可能原因包括：模型版本不匹配，模型参数丢失或者损坏。加了一个False，加这个的原因是：是否严格匹配状态字典中的键和模型中的模块。如果设置为False，则可以忽略不匹配的键，只加载匹配的部分。在我们使用PyTorch中用于加载模型状态字典（state_dict）的方法调用。目前在复现一篇论文的代码，执行的时候报错。
解决 RuntimeError: Error(s) in loading state_dict for : Missing key(s) in state_dict
lei_qi的博客
07-09 1万+
使用torch.nn.DataParallel多卡训练模型之后，加载模型前也需要打开多卡读取模型。 我最近使用多卡训练了一个模型。保存的方式是state_dict的方式。 然后在加载模型的时候就一直出错。 raise RuntimeError(‘Error(s) in loading state_dict for {}:\n\t{}’.format( RuntimeError: Error(s) in loading state_dict for BertMultiClassification: Missi
解决RuntimeError: Error(s) in loading state_dict for ResNet: Missing key(s) in state_dict: “conv1.0...
热门推荐
qq_39691492的博客
07-08 1万+
解决RuntimeError: Error(s) in loading state_dict for ResNet: Missing key(s) in state_dict: "conv1.0.weights", "conv1.1.weights", "conv1.1.bias", ...
PyTorch加载模型时报错RuntimeError: Error(s) in loading state_dict for *****: Missing key(s) in state_dict:
mj412828668的博客
04-07 1366
原作者的代码里面没有断点续训，我增加了这个功能，同时也引入了更多的参数，保存模型时增加了epoch、net.state_dict()、optimizer.state_dict()、scheduler.state_dict()等信息。方法2、将字典键值中的module.替换掉，或把原模型pth文件的key打印与现在模型的key进行比较，手动的为模型加载参数。找到训练代码中的net = nn.DataParallel(net)，注释掉再重新训练。或者采用上述方法2去加载模型。
pytorch加载模型报错RuntimeError: Error(s) in loading state_dict for SSD:Missing key(s) in state_dict:...
qq_40502460的博客
03-26 3424
报错信息： RuntimeError: Error(s) in loading state_dict for SSD: Missing key(s) in state_dict: “backbone.vgg.0.weight”, “backbone.vgg.0.bias”, “backbone.vgg.2.weight”, “backbone.vgg.2.bias”, “backbone.vgg.5.weight”, “backbone.vgg.5.bias”, “backbone.vgg.7.weight
RuntimeError: Error(s) in loading state_dict for XXX
OneQuestionADay的博客
05-11 2372
用torch保存训练后的模型 torch.save(model.state_dict(), 'file.pkl') 重新导入用于测试 model.load_state_dict(torch.load( 'file.pkl')) model = model.cuda() 报错： RuntimeError: Error(s) in loading state_dict for XXX: Missing key(s) in state_dict: "conv1.0.conv.weight", "conv1.
RuntimeError: Error(s) in loading state_dict for
weixin_36670529的博客
04-24 1万+
1.在载入模型参数前加上： model = nn.DataParallel(model) 2.更改torch版本 部分原因是模型之间torch版本不匹配导致，加载使用模型时和训练模型时的环境不一致，可以重新安装torch 3.增加load_state_dict()属性 model.load_state_dict(checkpoint['state_dict'], strict=False) ...
RuntimeError: DataLoader worker (pid(s) 9528, 8320) exited unexpectedly
01-06
RuntimeError: DataLoader worker (pid(s) 9528, 8320) exited unexpectedly 去网上搜教程，说是在loader定义的时候多进程的参数指定的问题，我的loader定义如下 loader = Data.DataLoader( dataset=torch_dataset,...
RuntimeError: Cannot run the event loop while another loop is running(目前没有解决)
12-21
在Python的Tornado框架中，`RuntimeError: Cannot run the event loop while another loop is running` 是一个常见的错误，它表明你的代码试图在一个已经运行中的事件循环（Event Loop）内启动另一个事件循环。...
解决SQLSTATE[HY000]: General error: 1205 Lock wait timeout exceeded_runtimeerror怎么修复
01-08
在MySQL数据库操作中，"SQLSTATE[HY000]: General error: 1205 Lock wait timeout exceeded" 是一个常见的错误，它意味着在执行事务时，系统等待锁定资源的时间超过了预设的限制。这个错误通常发生在并发环境中，当...
大学生挑战杯-喜树根器官培养和抗癌物质喜树碱生成的研究.rar
07-26
大学生挑战杯-喜树根器官培养和抗癌物质喜树碱生成的研究.rar
b278视频及游戏管理平台-springboot+vue.zip（可运行源码+sql文件+）
07-26
视频及游戏管理平台是一个很好的项目，结合了后端（Spring Boot）和前端（Vue.js）技术，实现了前后端分离。 视频及游戏管理平台是一个很好的项目，结合了后端（Spring Boot）和前端（Vue.js）技术，实现了前后端分离。 视频及游戏管理平台是一个很好的项目，结合了后端（Spring Boot）和前端（Vue.js）技术，实现了前后端分离。 视频及游戏管理平台是一个很好的项目，结合了后端（Spring Boot）和前端（Vue.js）技术，实现了前后端分离。 视频及游戏管理平台是一个很好的项目，结合了后端（Spring Boot）和前端（Vue.js）技术，实现了前后端分离。 视频及游戏管理平台是一个很好的项目，结合了后端（Spring Boot）和前端（Vue.js）技术，实现了前后端分离。
大模型应用-为Ollma开发的简单的HTML网页UI应用-附项目源码-优质项目实战.zip
最新发布
07-26
大模型应用_为Ollma开发的简单的HTML网页UI应用_附项目源码_优质项目实战
基于JAVA局域网监听软件的设计与开发(源代码+论文).rar
07-26
基于JAVA局域网监听软件的设计与开发(源代码+论文).rar
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"错误的方法。
评论 14
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值