tensorflow查看checkpoint变量值

作者: 小啾Kris | 来源:发表于2019-08-29 14:58 被阅读0次

tensorflow查看checkpoint变量值
如何输出查看tensorflow内的变量值
Tensorflow模块：tf.train.Checkpoint
2018-04-04
tensorflow保存checkpoint
记一次flink不做checkpoint的问题
2_checkpoints
cptn mobilenetv2 机器学习模型体积超小
tensorflow的升级与降级
杂记：Python中的小语法点

import tensorflow as tf

from tensorflow.python import pywrap_tensorflow

#使用tensorflow自带的pywrap_tensorflow读取模型

model_reader = pywrap_tensorflow.NewCheckpointReader(r"model.ckpt")

#变换成类似于dict形式的数据

var_dict = model_reader.get_variable_to_shape_map()

#最后，循环打印输出

for key in var_dict:

    print("variable name: ", key)

    print(model_reader.get_tensor(key))

tensorflow查看checkpoint变量值
如何输出查看tensorflow内的变量值
可查看博客如何输出查看tensorflow内的变量值
Tensorflow模块：tf.train.Checkpoint
tf.train.Checkpoint ：变量的保存与恢复 Tensorflow的Checkpoint机制将可...
2018-04-04
TensorFlow 到底有几种模型格式？ CheckPoint(*.ckpt)在训练 TensorFlow 模型...
tensorflow保存checkpoint
tf.train.Saver().save(sess, 'ckpts/') 在ckpts/ 路径下主要保存四个文件...
记一次flink不做checkpoint的问题
问题现象：Flink UI界面查看checkpoint的metrics发现一直没有做checkpoint，仔细排查...
2_checkpoints
tensorflow提供两种模型格式 checkpoint:依赖于创建模型的代码 SavedModel:与模型代码...
cptn mobilenetv2 机器学习模型体积超小
去年用tensorflow版本的ctpn跑过一些demo，效果不错，但是checkpoint在300MB左右，因为...
tensorflow的升级与降级
0、如何查看当前tensorflow版本： python import tensorflow as tf ...
杂记：Python中的小语法点
Tensorflow 类型 ▲Tensorflow：查看global variable和local variabl...