pytorch保存模型遇到点问题
Posted zawo1995
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pytorch保存模型遇到点问题相关的知识，希望对你有一定的参考价值。
今天用pytorch保存模型时遇到bug
Can‘t pickle <class ‘torch._C._VariableFunctions‘>
在google上查找原因，发现是保存时保存了整个模型的原因，而模型中有一些自定义的参数
将torch.save(model,save_path)改为torch.save(model.state_dict(),save_path)
然后载入模型也做相应的更改就好了
参考链接https://github.com/pytorch/pytorch/issues/7545
以上是关于pytorch保存模型遇到点问题的主要内容，如果未能解决你的问题，请参考以下文章