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('params.pkl'))
聚焦前沿AI与大模型技术探索，汇聚开发者及爱好者，共享开源项目、学习资源与行业资讯。
更多推荐
33
0- 0
已为社区贡献2条内容
所有评论(0)