pytorch学习笔记（五）：保存和加载模型

# 保存和加载整个模型torch.save(model_object, 'model.pkl')model = torch.load('model.pkl')# 仅保存和加载模型参数(推荐使用)torch.save(model_object.state_dict(), 'params.pkl')model_object.load_state_dict(torch.load('par...

u012436149

120199人浏览 · 2017-04-01 23:21:08

u012436149  ·  2017-04-01 23:21:08 发布

# 保存和加载整个模型
torch.save(model_object, 'model.pkl')
model = torch.load('model.pkl')

# 仅保存和加载模型参数(推荐使用)
torch.save(model_object.state_dict(), 'params.pkl')
model_object.load_state_dict(torch.load('params.pkl'))

AI大模型社区

聚焦前沿AI与大模型技术探索，汇聚开发者及爱好者，共享开源项目、学习资源与行业资讯。

加入社区

更多推荐

• · 不到 5 分钟，我用 Amazon Q 写了一个扫雷游戏！
• · 【AI大模型】Transformer架构到底是什么？
• · 大语言模型（LLM）面试前50常见问题英中对照版（附PDF下载）

不到 5 分钟，我用 Amazon Q 写了一个扫雷游戏！

AI大模型社区

【AI大模型】Transformer架构到底是什么？

AI大模型社区

大语言模型（LLM）面试前50常见问题英中对照版（附PDF下载）

AI大模型社区

欢迎加入社区

取消 确定