torch保存加载模型
Posted rise0111
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了torch保存加载模型相关的知识，希望对你有一定的参考价值。
保存模型
torch.save(my_model.state_dict(), "params.pkl")加载模型
先初始化model网络结构 model.load_state_dict(torch.load("params.pkl"))
以上是关于torch保存加载模型的主要内容，如果未能解决你的问题，请参考以下文章