Unexpected key(s) in state_dict

最新推荐文章于 2023-11-17 09:09:03 发布

小乙坛坛

最新推荐文章于 2023-11-17 09:09:03 发布

阅读量6.1k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： bug记录文章标签： Python

本文链接：https://blog.youkuaiyun.com/weixin_43487041/article/details/89325977

在训练多层神经网络后进行测试时遇到'Unexpected key(s) in state_dict'错误。问题源于测试模型的层数与训练模型不匹配。调整测试模型的层数以匹配训练模型，确保网络结构一致，即可解决此问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

训练多层网络后测试

报错.

RuntimeError: Error(s) in loading state_dict for Net_Atv:
	Unexpected key(s) in state_dict: "net.30.conv1.weight",
	 "net.30.conv1.bias", "net.30.conv2.weight", "net.30.conv2.bias", "net.30.conv3.weight", "net.30.conv3.bias", "net.30.conv4.weight",
	  "net.30.conv4.bias", "net.31.conv1.weight", "net.31.conv1.bias", "net.31.conv2.weight",