当前位置: article > 正文

【pytorch加载模型错误】Unexpected key(s) in state_dict: “epoch“, “arch“, “state_dict“, “optimizer“, “args“_unexpected key(s) in state_dict: "model", "optimiz

作者：我家小花儿 | 2024-02-28 16:17:29

踩

unexpected key(s) in state_dict: "model", "optimizer", "lr_scheduler", "epoc

方案1.
尝试用

state_dict = torch.load(model_name)
model.load_state_dict({k.replace('module.', ''): v for k, v in state_dict.items()})
1
2

方案2.
训练好的模型文件好像字典键值有很多个，如optimizer，epoch，args等，我们只需要模型参数文件state_dict

state_dict = torch.load(new_model)
run_logger.info('{} start evaluation!!!'.format(new_model))
model.load_state_dict(state_dict['state_dict'])
1
2
3

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家小花儿/article/detail/160095