“torch.load“中出现的“Unexpected key(s) in state_dict“报错问题

问题:

解决:

添加strict=False,允许加载过程中出现不匹配的键。但请注意,仍然需要确保模型中的主要参数能够正确加载,以确保模型的有效性。

python 复制代码
model.load_state_dict(state_dict)
# 改为:
model.load_state_dict(state_dict, strict=False)
相关推荐
WXX_s12 分钟前
【OpenCV篇】OpenCV——03day.图像预处理(2)
人工智能·python·opencv·学习·计算机视觉
Jackilina_Stone1 小时前
【论文|复现】YOLOFuse:面向多模态目标检测的双流融合框架
人工智能·python·目标检测·计算机视觉·融合
双叶8362 小时前
(Python)文件储存的认识,文件路径(文件储存基础教程)(Windows系统文件路径)(基础教程)
开发语言·windows·python
枫昕柚2 小时前
python
开发语言·python
木头左2 小时前
自动驾驶领域中的Python机器学习
python·机器学习·自动驾驶
Dxy12393102162 小时前
Python Requests-HTML库详解:从入门到实战
开发语言·python·html
seasonsyy2 小时前
2.安装CUDA详细步骤(含安装截图)
python·深度学习·环境配置·cuda
写点什么啦3 小时前
一键修复ipynb,Jupyter Notebook损坏文件
ide·python·jupyter
fishwheel3 小时前
Android:Reverse 实战 part 2 番外 IDA python
android·python·安全
测试19983 小时前
cmake应用:集成gtest进行单元测试
自动化测试·软件测试·python·测试工具·职场和发展·单元测试·测试用例