pytorch 保存模型出错

今天在使用pytorch保存模型时，突然出现Can’t pickle typing.Union[torch.Tensor, NoneType]: it’s not the same object的错误，找了好多文章也没找到解决办法，后来把保存代码由save(model,path)换成save(model.state_dict(), path)终于不报错了...

江_小_白

1502人浏览 · 2021-07-18 17:15:01

江_小_白 · 2021-07-18 17:15:01 发布

今天在使用pytorch保存模型时，突然出现
Can’t pickle typing.Union[torch.Tensor, NoneType]: it’s not the same object
的错误，找了好多文章也没找到解决办法，后来把保存代码由
save(model,path)换成save(model.state_dict(), path)终于不报错了