Pytorch：Unexpected key(s) in state_dict:

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

wchzh2015

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

阅读量1.6w

点赞数 3

CC 4.0 BY-SA版权

分类专栏： pytorch

本文链接：https://blog.youkuaiyun.com/wchzh2015/article/details/85254215

pytorch 专栏收录该内容

4 篇文章

订阅专栏

本文探讨了在PyTorch环境下，使用交叉验证后重新加载并求平均多个模型结果时遇到的问题及解决办法。具体涉及PyTorch0.4版本与Python3.6环境下，针对ResNet101模型的加载过程进行调整，以确保模型能够正确加载并执行后续操作。

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

实验环境
1. pytorch 0.4
2. Python 3.6
在使用交叉验证后，将保存的模型重新加载求平均的结果。但是出现了以下的错误。
实验的bug截图

在此感谢这篇博文：https://blog.youkuaiyun.com/jacke121/article/details/82495184
自己修改后的关键代码如下：

model_resnet101 = get_net()
model_resnet101.cuda()
model_resnet101.load_state_dict({k.replace('module.',''):v for k,v in torch.load("densenet169_rnn_fold_1_model_best_f1.pth.tar")['state_dict'].items()})