tensorflow学习之——从保存的权重中使用tensorboard查看网络

本文介绍了如何使用TensorFlow的Saver保存模型,并通过TensorBoard来可视化网络结构。主要文件包括checkpoint、.meta、.index和.data文件,分别存储模型状态、网络结构和权重。在应用中,可以加载他人训练好的模型,通过TensorBoard查看网络结构。

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

0. 写作目的

好记性不如烂笔头。

1. tensorboard模型保存说明

采用tf.train.Saver保存的模型,主要文件有:

1) checkpoint : 用于记录训练过程中保存的各模型,在第一行会保存最新的模型的路径: model.ckpt

2)  model.ckpt.meta: 为保存的Graph,即网络

3) model.ckpt.index : 为string-string table,table的key值为tensor名,value为BundleEntryProto.

4) model.ckpt.data-00000-of-00001: 为Graph中的权重

2. tensorboard可视化网络

1.1 应用场景

对于别人训练好的tensorboard模型,包含的文件如1中所示。

然后加载图,并查看网络的结构。

modelDir = currentDir + '/model'
summariesDir =  currentDir + '/tensorboard'
def test_main():
    with tf.Session() as sess:
        model_file = tf.train.latest_checkpoint(modelDir
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值