Missing key(s) in state_dict

Missing key(s) in state_dict | unexpected_keys | 模型加载 [torch.load] 报错【解决方案】

出现这种情况基本上是因为模型网络与加载的模型权重不匹配造成的

当你出现这种错误时，建议先去看看自己的权重数据是怎么样的（这里以.pth格式的权重数据为例）：

• 加载权重数据：ckpt = torch.load("yolox_m.pth", map_location=torch.device('cpu'))

• print(ckpt.keys()) # keys()

• print(ckpt['model']) # 之后有其他需求比如要看 key 为 model 的内容有啥

这里的权重信息一般在‘model’这个key中，所以你从ckpt中导入权重数据，就会发生错位，导致权重数据不匹配。

所以由此在将权重数据加载到网络中去时，应该导入ckpt权重数据中的‘model’键：

model.load_state_dict(ckpt['model'])