pytorch利用torch.save()保存模型及相关参数,以及读取保存之后的文件
2019-03-28 09:40
841 查看
转自:https://www.cnblogs.com/qinduanyinghua/p/9311410.html
假设网络为model = Net(), optimizer = optim.Adam(model.parameters(), lr=args.lr), 假设在某个epoch,我们要保存模型参数,优化器参数以及epoch
一、
1. 先建立一个字典,保存三个参数:
state = {‘net':model.state_dict(), 'optimizer':optimizer.state_dict(), 'epoch':epoch}
2.调用torch.save():
torch.save(state, dir)
其中dir表示保存文件的绝对路径+保存文件名,如'/home/qinying/Desktop/modelpara.pth'
二、
当你想恢复某一阶段的训练(或者进行测试)时,那么就可以读取之前保存的网络模型参数等。
checkpoint = torch.load(dir)
model.load_state_dict(checkpoint['net'])
optimizer.load_state_dict(checkpoint['optimizer'])
start_epoch = checkpoint['epoch'] + 1
相关文章推荐
- pytorch保存模型等相关参数,利用torch.save(),以及读取保存之后的文件
- 读取属性文件,保存相关配置参数
- OpenGL读取obj模型 源码 以及文件解析
- java实现保存文件图片以及读取
- tensorflow: 保存和加载模型, 参数;以及使用预训练参数方法
- Azureus研究-配置文件的参数读取和参数保存
- asp.net读取Xml文件到DataSet,以及DataSet保存为Xml,利用自带的强大功能
- TensorFlow模型文件保存和读取
- JsonResult序列化并保存json文件 以及对json文件读取反序列
- 《tensorflow实战笔记》对于TED(en-zh)数据集进行Seq2Seq模型实战,以及对应的Attention机制(tf保存模型读取模型)
- 流和文件:保存和读取字符级数据以及随机访问
- c语言 实现按行读取文件。把内容按照第三种内存模型打包数据传出,把行数通过函数参数传出
- 基于卷积神经网络的数据预处理以及模型参数保存
- 读取修改caffemodel文件里的网络模型参数
- 浅谈在vue中用webpack打包之后运行文件的问题以及相关配置方法
- 软件保存复杂参数到文件的一个实例,容器相关
- python读取Excel文件以及保存Excel文件
- keras:保存keras学习好的深度神经网络模型参数为二进制和txt文件
- MFC环境下搭建OpenGL框架并进行数据文件读取以及模型绘制的注意事项
- casperjs接收外部参数,以及读取本地文件