当前位置:   article > 正文

model.load_state_dict(checkpoint[‘state_dict‘]) KeyError: ‘state_dict‘_keyerror: 'state_dict

keyerror: 'state_dict

问题描述

加载模型时,报错:model.load_state_dict(checkpoint[‘state_dict’]) KeyError: ‘state_dict’

解决

1. 如果你的模型命名为 checkpoint.ckpt,试试将其改为checkpoint.pt
2. model.load_state_dict(torch.load(pth))
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/160093
推荐阅读
相关标签
  

闽ICP备14008679号