tensorflow模型保存, 单个文件, 模型读取, 部分模型保存,读取

本文介绍了如何在TensorFlow中保存和加载模型,包括完整模型的保存、导出为单个文件、部分模型的保存与加载。详细步骤包括生成保存文件、转换变量为常量、节点定位以及在TensorBoard中进行可视化。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

保存完整的模型

生成一下4个文件:

checkpoint
.meta
.data-00000-of-00001
.index

使用此方法保存时,会生成3个文件,其中后缀为.meta保存图的结构和常量.data.index保存模型的权重,偏差,梯度其他的变量的所有的值等。

  • 创建服务对象,如果Saver()有传入,表示只对传入的值有相应的后续效果

    saver = tf.train.Saver()
    
  • 适应saver进行模型保存

    saver.save(sess,保存的文件名)
    

导出模型为一个文件

将图中所有的变量装换为常量tf.graph_util.convert_variables_to_constants()

写入到一个模型中tf.train.writer_graph()

import tensorflow as tf

w1 = tf.Variable(20.0, name="w1")
w2 = tf.Variable(30.0, name="w2")
b1= tf.Variable(2.0,name=
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值