同一个模型,pytorch加载参数文件时报错:missing key(s) in state_dict

问题描述

使用pytorch保存模型的state_dict为pt文件后,同一个模型结构加载pt文件,
在这里插入图片描述
显示缺失一部分的key:
同一个模型,pytorch加载参数文件时报错:missing key(s) in state_dict_第1张图片
同时,显示多出了一部分key:
同一个模型,pytorch加载参数文件时报错:missing key(s) in state_dict_第2张图片

原因分析和解决方案:

可能因为使用了并行计算,参数文件的key命名不一致,修改如下:
在这里插入图片描述

你可能感兴趣的:(pytorch)