技术栈

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

代码小白的成长2023-10-17 20:48

问题:

解决:

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

python 复制代码
model.load_state_dict(state_dict)
# 改为:
model.load_state_dict(state_dict, strict=False)
上一篇:MAT查找类(岔路口)-技巧
下一篇:DIN模型和SIM模型原理与实践
相关推荐
思则变
1 小时前
[Pytest] [Part 2]增加 log功能
开发语言·python·pytest
漫谈网络
2 小时前
WebSocket 在前后端的完整使用流程
javascript·python·websocket
try2find
3 小时前
安装llama-cpp-python踩坑记
开发语言·python·llama
博观而约取
4 小时前
Django ORM 1. 创建模型(Model)
数据库·python·django
精灵vector
6 小时前
构建专家级SQL Agent交互
python·aigc·ai编程
Zonda要好好学习
6 小时前
Python入门Day2
开发语言·python
Vertira
6 小时前
pdf 合并 python实现(已解决)
前端·python·pdf
太凉
6 小时前
Python之 sorted() 函数的基本语法
python
项目題供诗
6 小时前
黑马python(二十四)
开发语言·python
晓1313
7 小时前
OpenCV篇——项目(二)OCR文档扫描
人工智能·python·opencv·pycharm·ocr
热门推荐
01手机电脑之间快速传输图片视频文件,不压缩画质、不限制大小的方法!02如何将markdown生成pdf03Coze扣子平台完整体验和实践(附国内和国际版对比)04Coze实战第13讲:飞书多维表格读取+豆包生图模型,轻松批量生成短剧封面05鸽了六年的某大厂面试题:你会手写一个模板引擎吗?06扣子(coze)实战|我用扣子搭建了一个自动分析小红薯笔记内容的AI应用|详细步骤拆解07免费可用!最强AI数字人对口型神器:让照片开口说话唱歌,支持多人对口型+全身动作,1分钟学会!(附保姆级教程)08《C#图解教程 第5版》深度推荐09DeepSeek各版本说明与优缺点分析10C#调用WechatOCR.exe实现本地OCR文字识别