技术栈

“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模型原理与实践
相关推荐
ValhallaCoder
2 小时前
hot100-二叉树I
数据结构·python·算法·二叉树
猫头虎
3 小时前
如何排查并解决项目启动时报错Error encountered while processing: java.io.IOException: closed 的问题
java·开发语言·jvm·spring boot·python·开源·maven
八零后琐话
4 小时前
干货:程序员必备性能分析工具——Arthas火焰图
开发语言·python
青春不朽512
5 小时前
Scrapy框架入门指南
python·scrapy
MZ_ZXD001
5 小时前
springboot旅游信息管理系统-计算机毕业设计源码21675
java·c++·vue.js·spring boot·python·django·php
全栈老石
6 小时前
Python 异步生存手册:给被 JS async/await 宠坏的全栈工程师
后端·python
梨落秋霜
6 小时前
Python入门篇【模块/包】
python
阔皮大师
7 小时前
INote轻量文本编辑器
java·javascript·python·c#
小法师爱分享
7 小时前
StickyNotes,简单便签超实用
java·python
热门推荐
01GitHub 镜像站点02Claude Code + GLM4.7 避坑指南:解决 Unable to connect to Anthropic services03openclaw配置教程(linux+局域网ollama)04UV安装并设置国内源05OpenClaw Chrome扩展使用教程 - 浏览器中继控制06Linux下V2Ray安装配置指南07Claude Code Skills 实用使用手册08Vue-skills的中文文档09一文了解国产算子编程语言 TileLang,TileLang 对国产开源生态的影响与启示10让 Trae IDE 智能体 “读懂”文档 Excel+PDF+DOCX :mcp-documents-reader 工具使用指南