TensorFlow 模型持久化

为了使训练模型可以复用，需要将训练得到的神经网络模型持久化。Tensorflow通过tf.train.Saver类实现模型的保存和还原。

Tensorflow的模型一般会保存在后缀为.ckpt的文件中， checkpoint 保存目录下所有的模型文件列表。

# 1. 保存计算两个变量和的模型。
v1 = tf.Variable(tf.random_normal([1], stddev=1, seed=1))
v2 = tf.Variable(tf.random_normal([1], stddev=1, seed=1))
result = v1 + v2

init_op = tf.global_variables_initializer()
saver = tf.train.Saver()

with tf.Session() as sess:
    sess.run(init_op)
    saver.save(sess, "Saved_model/model.ckpt")
    
# 2. 加载保存了两个变量和的模型。
with tf.Session() as sess:
    saver.restore(sess, "Saved_model/model.ckpt")
    print(sess.run(result))

Tensorflow支持保存或加载部分变量，比如一个已训练好的8层神经网络，想在此基础上尝试训练一个新的10层神经网络，可以将前面8层网络的参数直接加载到模型中，仅对最后两层神经网络进行训练。

为保存或加载部分变量，在声明tf.train.Saver类时可以提供一个列表来指定需要保存或加载的变量。Tensorflow通过字典将模型保存时的变量名和需要加载的变量联系起来。