第一种 保存整个模型文件
# 保存和加载整个模型
torch.save(model_object, 'model.pth')
model = torch.load('model.pth')
第二种 仅保存模型的权重
# 仅保存和加载模型参数(推荐使用)
torch.save(model_object.state_dict(), 'weights.pth')
model.load_state_dict(torch.load('weights.pth'))
pytorch保存模型、加载模型的两种方式
2021-11-18
206
版权
版权声明：
本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《
阿里云开发者社区用户服务协议》和
《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写
侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。
简介：
pytorch保存模型、加载模型的两种方式
目录
相关文章
|
3天前
|
机器学习/深度学习
PyTorch
算法框架/工具
【从零开始学习深度学习】28.卷积神经网络之NiN模型介绍及其Pytorch实现【含完整代码】
【从零开始学习深度学习】28.卷积神经网络之NiN模型介绍及其Pytorch实现【含完整代码】
11
2
2
|
3天前
|
机器学习/深度学习
PyTorch
算法框架/工具
【从零开始学习深度学习】26.卷积神经网络之AlexNet模型介绍及其Pytorch实现【含完整代码】
【从零开始学习深度学习】26.卷积神经网络之AlexNet模型介绍及其Pytorch实现【含完整代码】
11
3
3
|
3天前
|
机器学习/深度学习
自然语言处理
PyTorch
【从零开始学习深度学习】48.Pytorch_NLP实战案例：如何使用预训练的词向量模型求近义词和类比词
【从零开始学习深度学习】48.Pytorch_NLP实战案例：如何使用预训练的词向量模型求近义词和类比词
9
0
0
|
3天前
|
机器学习/深度学习
算法
PyTorch
【从零开始学习深度学习】44. 图像增广的几种常用方式并使用图像增广训练模型【Pytorch】
【从零开始学习深度学习】44. 图像增广的几种常用方式并使用图像增广训练模型【Pytorch】
5
0
0
|
3天前
|
机器学习/深度学习
PyTorch
算法框架/工具
【从零开始学习深度学习】29.卷积神经网络之GoogLeNet模型介绍及用Pytorch实现GoogLeNet模型【含完整代码】
【从零开始学习深度学习】29.卷积神经网络之GoogLeNet模型介绍及用Pytorch实现GoogLeNet模型【含完整代码】
11
1
1
|
3天前
|
机器学习/深度学习
PyTorch
算法框架/工具
【从零开始学习深度学习】27.卷积神经网络之VGG11模型介绍及其Pytorch实现【含完整代码】
【从零开始学习深度学习】27.卷积神经网络之VGG11模型介绍及其Pytorch实现【含完整代码】
7
2
2
|
3天前
|
机器学习/深度学习
算法
PyTorch
【从零开始学习深度学习】25.卷积神经网络之LeNet模型介绍及其Pytorch实现【含完整代码】
【从零开始学习深度学习】25.卷积神经网络之LeNet模型介绍及其Pytorch实现【含完整代码】
8
2
2
|
3天前
|
机器学习/深度学习
并行计算
PyTorch
|
3天前
|
存储
机器学习/深度学习
PyTorch
【从零开始学习深度学习】19. Pytorch中如何存储与读取模型：torch.save、torch.load与state_dict对象
【从零开始学习深度学习】19. Pytorch中如何存储与读取模型：torch.save、torch.load与state_dict对象
13
2
2
|
3天前
|
机器学习/深度学习
PyTorch
算法框架/工具
热门文章
最新文章
1
【Pytorch基础教程27】DeepFM推荐算法
2
PyTorch 小课堂！带你解析数据处理全流程（二）
3
GitHub深度学习框架榜单：PyTorch快速增长(2/11-4/12)
4
ResNet代码复现＋超详细注释（PyTorch）
5
Pytorch 各种奇葩古怪的使用方法
6
如何用PyTorch处理人脸姿态的数据？
7
PyTorch宣布支持苹果M1芯片GPU加速：训练快6倍，推理提升21倍
8
pytorch安装、环境搭建及在pycharm中的设置
9
基于Pytorch运行中出现RuntimeError: Not compiled with CUDA support此类错误解决方案
10
Pytorch学习笔记-04 权值初始化与损失函数
1
探索PyTorch：介绍及常用工具包展示
30
2
pytorch实战---IMDB情感分析
142
3
使用PyTorch解决多分类问题：构建、训练和评估深度学习模型
150
4
使用PyTorch加载数据集：简单指南
95
5
使用PyTorch处理多维特征输入的完美指南
90
6
用PyTorch轻松实现二分类：逻辑回归入门
66
7
从零开始学习线性回归：理论、实践与PyTorch实现
32
8
PyTorch实例：简单线性回归的训练和反向传播解析
52
9
Pytorch中张量的高级选择操作
46
10
【AMP实操】解放你的GPU运行内存！在pytorch中使用自动混合精度训练
95