pytorch错误：Missing key(s) in state_dict、Unexpected key(s) in state_dict解决
2024-03-05 02:12:00
版权声明：本文为博主原创文章，欢迎转载，并请注明出处。联系方式：460356155@qq.com
在模型训练时加上：
model = nn.DataParallel(model)
cudnn.benchmark = True
可以加速训练速度。
但是需要注意的是，训练后保存的模型参数在被加载到模型前，需要对模型加上：
model = nn.DataParallel(model)
cudnn.benchmark = True
否则加载时会出现错误：
RuntimeError: Error(s) in loading state_dict for ResNet:
Missing key(s) in state_dict: xxxxxxxx
Unexpected key(s) in state_dict: xxxxxxxxxx
最新文章
- app开发外包注意事项，2017最新资讯
- NHibernate之映射文件配置说明
- iptables 开启端口
- tornado 学习笔记8 模板以及UI
- 循序渐进Python3（十）-- 4 -- paramiko
- 理解 OpenStack 高可用(HA) （4）: Pacemaker 和 OpenStack Resource Agent （RA）
- Java消息服务
- 业余草教你解读Spark源码阅读之HistoryServer
- 为啥REST如此重要？
- Day7 面向对象和类的介绍
- c#操作SQL Server入门总结
- MyBatis映射文件3(参数处理Map)
- angualrjs 文本框去掉表情
- sql随机查询数据语句(NewID(),Rnd,Rand(),random())
- 升级 Centos 6.5/6.7 的 php 版本
- C语言学习感受
- 力扣（LeetCode）976. 三角形的最大周长
- 2Q - Fibbonacci Number
- python 字符串的split()函数详解
- iOS in-app purchase详解