【PyTorch】模型加载与保存

最新推荐文章于 2023-12-11 20:42:16 发布

北境の守卫

最新推荐文章于 2023-12-11 20:42:16 发布

阅读量620

点赞数

分类专栏： PyTorch 文章标签： pytorch model save load

本文链接：https://blog.csdn.net/baishuo8/article/details/90237546

版权

本文详细介绍了如何在PyTorch中保存和加载模型，包括从CPU到CPU，GPU到GPU，以及GPU到CPU的转换操作，为模型的训练和部署提供了清晰的指导。

摘要由CSDN通过智能技术生成

Backto PyTorch Index

模型保存

# 保存整个模型的结构信息和参数信息, 保存的对象是模型 model ，可以是pth方式，也可以是pkl的方式
torch.save(model, './model.pth')   # 方式1
# 保存模型的参数, 保存的对象是模型的状态 model.state_dict()
torch.save(model.state_dict(), './model_state.pth'