模型加载权重不匹配

最新推荐文章于 2024-06-04 08:35:52 发布

Pochacc_ZZ

最新推荐文章于 2024-06-04 08:35:52 发布

阅读量1.9k

点赞数 1

文章标签： python 开发语言

本文链接：https://blog.csdn.net/weixin_46832764/article/details/128262930

版权

问题描述：

The model and loaded state dict do not match exactly.

RuntimeError: Error(s) in loading state_dict for MLMemAESC:Missing key(s) in state_dict.

解决方案：

checkpoint=checkpoint['state_dict']
keys = []
#remove extra module
for k, v in checkpoint.items():
     if k.startswith('back'):
            continue
            keys.append(k)
new_dict = {k: checkpoint[k] for k in keys}
  
#remove extra key
from collections import OrderedDict
new_dict1 = OrderedDict()
for i, j in new_dict.items():
    name = i[5:]
    new_dict1[name] = j
head_model.load_state_dict(new_dict1)