pytorch保存模型记录

最新推荐文章于 2024-04-16 00:15:47 发布

lowl

最新推荐文章于 2024-04-16 00:15:47 发布

阅读量1.1k

点赞数 1

分类专栏： python 文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/weixin_48629412/article/details/121269458

版权

python 专栏收录该内容

19 篇文章 0 订阅

订阅专栏

这篇博客讨论了在PyTorch中遇到的模型加载错误，如'Error(s) in loading state_dict'。问题一指出网络结构不匹配，导致缺少和意外的键。解决方法是在load_state_dict时设置strict=False，允许键名不一致。问题二中提到即使使用此方法，依然存在'fc'层的键不匹配问题。

摘要由CSDN通过智能技术生成

RuntimeError: Error(s) in loading state_dict

情况一：

原因：网络结构明显不一致

Missing key(s) in state_dict: "weight", "bias". 
        Unexpected key(s) in state_dict: "conv1.weight", "bn1.weight", "bn1.bias", "bn1.running_mean", "bn1.running_var", "bn1.num_batches_tracked", "layer1.0.conv1.weight", "layer1.0.bn1.weight", "layer1.0.bn1.bias", "layer1.0.bn1.running_mean", "layer1.0.bn1.running_var", "layer1.0.bn1.num_batches_tracked", "layer1.0.conv2.weight", "layer1.0.bn2.weight", "layer1.0.bn2.bias", "layer1.0.bn2.running_mean", "layer1.0.bn2.running_var", "layer1.0.bn2.num_batches_tracked", "layer1.0.conv3.weight", "layer1.0.bn3.weight", "layer1.0.bn3.bias", "layer1.0.bn3.running_mean", "layer1.0.bn3.running_var", "layer1.0.bn3.num_batches_tracked", "layer1.0.downsample.0.weight", "layer1.0.downsample.1.weight", "layer1.0.downsample.1.bias", "layer1.0.downsample.1.running_mean", "layer1.0.downsample.1.running_var", "layer1.0.downsample.1.num_batches_tracked", "layer1.1.conv1.weight", "layer1.1.bn1.weight", "layer1.1.bn1.bias", "layer1.1.bn1.running_mean", "layer1.1.bn1.running_var", "layer1.1.bn1.num_batches_tracked", "layer1.1.conv2.weight", "layer1.1.bn2.weight", "layer1.1.bn2.bias", "layer1.1.bn2.running_mean", "layer1.1.bn2.running_var", "layer1.1.bn2.num_batches_tracked", "layer1.1.conv3.weight",

情况二：

解决方法：load_state_dict中加入strict=False，强制要求key名称可以不一致

load_state_dict(strict=False)

RuntimeError: Error(s) in loading state_dict for Linear:
        Missing key(s) in state_dict: "weight", "bias". 
        Unexpected key(s) in state_dict: "fc.weight", "fc.bias".