模型的保存使用了函数tf.train.Saver()
通过saver.saver(sess, ”path“ )保存
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.initialize_all_variables()
#声明tf.train.Saver类
saver=tf.train.Saver()
with tf.Session() as sess:
sess.run(init_op)
saver.save(sess,"/home/kaitang/test/model/model.ckpt")
文件下
出现了 4个文件
checkpoint
model.ckpt.data-00000-of-00001
model.ckpt.index
model.ckpt.meta