Pytorch加载模型时报错

加载模型时出现Unexpected key(s) in state_dict错误 报错截图如下: 反复排查问题没发现为何如此,查看pytorch中文文档发现保存和加载模型方法都完全正确,模型保存和加载代码对比中文文档截图如下: 其中一个方法是在加载模型时添加参数strict=False,可以只保留键值相同的参数避免出错,用法如下:model.load_state_dict(ckpt[‘state_
相关文章
相关标签/搜索