pytorch错误：Missing key(s) in state_dict、Unexpected key(s) in state_dict解决

2024年1月13日 574次阅读来源: pytorch

在模型训练时加上：

model = nn.DataParallel(model)
cudnn.benchmark = True
可以加速训练速度。
但是需要注意的是，训练后保存的模型参数在被加载到模型前，需要对模型加上：

model = nn.DataParallel(model)
cudnn.benchmark = True

否则加载时会出现错误：
RuntimeError: Error(s) in loading state_dict for ResNet:
    Missing key(s) in state_dict: xxxxxxxx
　　Unexpected key(s) in state_dict: xxxxxxxxxx

    原文作者：pytorch
    原文地址: https://www.cnblogs.com/zhengbiqing/p/10434704.html
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。