这个屌丝很懒，什么也没留下！

pytorch学习笔记（五）：保存和加载模型_model = torch.load('model.pkl')

作者：我家小花儿 | 2024-04-07 11:35:21

model = torch.load('model.pkl')

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

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

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家小花儿/article/detail/378353