1. 程式人生 > >5.4 TensorFlow模型持久化

5.4 TensorFlow模型持久化

1.模型的儲存

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()
saver = tf.train.Saver()
with tf.Session() as sess:
	sess.run(init_op)
	saver.save(sess,"model/model.ckpt")

上面的程式碼共有三個變數 result = v1 + v2

使用了一個核心類 tf.train.Saver()

生成了四個檔案checkpoint,model.ckpt.meta和model.ckpt.data-00000-of-00001,model.ckpt.index

2.模型的