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