Mmagic1的博客

Tensorflow训练的模型，常常需要保存下来，便于调用，有两种方式保存：ckpt和pb 1、保存为ckpt模型，代码为： import tensorflow as tf v1 = tf.Variable(tf.constant(1.0,shape = [1],name = "v1")) v2 = tf.Variable(tf.constant(2.0,shape = [1],name = "v2")) result = v1 +v2 init_op = tf.global_variables

一、查看ckpt图结构 1、在ckpt文件所在文件夹中新建check_ckpt.py文件，代码如下 import tensorflow as tf from tensorflow.summary import FileWriter sess = tf.Session() tf.train.import_meta_graph("./model.ckpt.meta") FileWriter("__tb", sess.graph) 2、运行check_ckpt.py，生成__tb文件夹 3、在__tb