模型保存与加载|学习笔记

简介: 快速学习模型保存与加载

开发者学堂课程【深度学习框架 TensorFlow 入门模型保存与加载学习笔记,与课程紧密联系,让用户快速学习知识。

课程地址https://developer.aliyun.com/learning/course/773/detail/13550


模型保存与加载


内容介绍:  

一、模型的保存与加载

二、实现代码


一、模型的保存与加载

tf.train.Saver(var_list= None,max.to_keep=5)

。保存和加载模型(保存文件格式: checkpoint 文件)

。var_list:指定将要保存和还原的变量,它可以作为-个dict或一个列表传递.

。max_to_keep:指示要保留的最近检查点文件的最大数量。创建新文件时,会删除较旧的文件。如果无或0,则保留所有检查点文件。默认为5 (即保留最新的5个检查点文件。)

使用

例如:

指定目录+模型名字

saver.save(sess,'/tmp/ckpt/test/myregression.ckpt')

saver.restore(sess,’/tmp/ckpt/test/myregression.ckpt')

如要判断模型是否存在,直接指定目录

checkpoint = tf.train.latest_checkpoint("./tmp/modeL/")

saver.restore(sess,checkpoint )


二、实现代码

#创建 Saver 对象,

Saver=tf.train.Saver()

#保存模型

if i % 10==0:

saver.save(sess,"./tmp/modeL/my_linear.ckpt")

#加载模型

if os.path.exists("./tmp/modeL/checkpoint"):

saver.restore(sess,"./tmp/modeL/my_Linear.ckpt")

print(“训练后模型参数为:权重%f,偏置%f,损失为%f" % (weights. eval(), bias.eval(), error.eval()))

相关文章
|
2月前
|
移动开发 TensorFlow 算法框架/工具
只保存和加载网络权重
【8月更文挑战第21天】只保存和加载网络权重。
27 2
|
2月前
|
机器学习/深度学习
加载CNN保存模型
【8月更文挑战第10天】加载CNN保存模型。
40 12
|
3月前
|
存储 计算机视觉 Python
加载图像,显示并保存
【7月更文挑战第28天】加载图像,显示并保存。
35 2
|
5月前
|
PyTorch 算法框架/工具 异构计算
pytorch 模型保存与加载
pytorch 模型保存与加载
36 0
|
5月前
|
机器学习/深度学习 PyTorch 调度
PyTorch进阶:模型保存与加载,以及断点续训技巧
【4月更文挑战第17天】本文介绍了PyTorch中模型的保存与加载,以及断点续训技巧。使用`torch.save`和`torch.load`可保存和加载模型权重和状态字典。保存模型时,可选择仅保存轻量级的状态字典或整个模型对象。加载时,需确保模型结构与保存时一致。断点续训需保存训练状态,包括epoch、batch index、optimizer和scheduler状态。中断后,加载这些状态以恢复训练,节省时间和资源。
|
Linux
模块的加载过程四
模块的加载过程四
118 0
|
存储 计算机视觉
三天学会opencv(一)——加载,修改,保存图像
三天学会opencv(一)——加载,修改,保存图像
187 0
三天学会opencv(一)——加载,修改,保存图像
一个简单的页面加载管理类(包含加载中,加载失败,数据为空,加载成功)(下)
在最近公布的比赛框架中,发现了页面加载管理类,觉得挺有用的,所以做个简单的笔记。
111 0
一个简单的页面加载管理类(包含加载中,加载失败,数据为空,加载成功)(下)
一个简单的页面加载管理类(包含加载中,加载失败,数据为空,加载成功)(上)
在最近公布的比赛框架中,发现了页面加载管理类,觉得挺有用的,所以做个简单的笔记。
94 0
一个简单的页面加载管理类(包含加载中,加载失败,数据为空,加载成功)(上)
|
PyTorch 算法框架/工具 Python
pytorch保存参数及模型的两种方式
pytorch保存参数及模型的两种方式
546 0