当前位置: 代码迷 >> 综合 >> RuntimeError: Error(s) in loading state_dict for Net:unexpected key(s) in state_dict: XXX
  详细解决方案

RuntimeError: Error(s) in loading state_dict for Net:unexpected key(s) in state_dict: XXX

热度:82   发布时间:2023-11-13 12:14:20.0

加载模型时发生错误RuntimeError: Error(s) in loading state_dict for Net:unexpected key(s) in state_dict: XXX
解决方法:

model.load_state_dict(torch.load('models/params.pt'),strict=False)

在参数中加入strict=False
此类错误一部分原因是pytorch环境不同导致的

  相关解决方案