Missing key(s) in state_dict: Unexpected key(s) in state_dict
2024-09-07 14:38:16
如果加载的预训练模型之前使用了torch.nn.DataParallel(),而此时的训练并没有使用,则会出现这样的错误。
解决方案有两个:
1:此时的训练加入torch.nn.DataParallel()即可。
2:创建一个没有module.的新字典,即将原来字典中module.删除掉。
解决方案1:
model = torch.nn.DataParallel(model)
cudnn.benchmark = True
解决方案2:
# original saved file with DataParallel
state_dict = torch.load('myfile.pth')
# create new OrderedDict that does not contain `module.`
from collections import OrderedDict
new_state_dict = OrderedDict()
for k, v in state_dict.items():
name = k[7:] # remove `module.`
new_state_dict[name] = v
# load params
model.load_state_dict(new_state_dict)
解决方案3:
model.load_state_dict({k.replace('module.',''):v for k,v in torch.load('myfile.pth').items()})
最新文章
- C# out ref 重载
- redis入门笔记(2)
- inuit.css – 基于 Sass 的强大,可扩展的 CSS 框架
- [Google Guava]学习--新集合类型BiMap
- FR #2题解
- asp.net 下载Excel (数据流,不保存)--客户端
- nyoj-291 互素数个数 欧拉函数
- [转] jQuery按键响应事件keypress对应的按键编码keycode
- docker 数据管理
- Docker - 用Flannel跨主机
- 怎样在Win7系统建立并开启Wifi热点
- 将Python程序打包为exe方法
- .Net Core微服务系列--理论篇
- C++基础——类继承中方法重载
- (转)A Recipe for Training Neural Networks
- 解决:fatal error LNK1104: 无法打开文件“libc.lib”
- iOS -- Effective Objective-C 阅读笔记 (2)
- 常见 HTTP/FTP/WebSocket 错误代码大全 - 转
- LB+nginx+tomcat7集群模式下的https请求重定向(redirect)后变成http的解决方案
- 大数据入门到精通2--spark rdd 获得数据的三种方法