在Tensorflow中,我们可以创建一个tf.train.Saver() 类的实例来保存训练完得到的参数:
'''
saver = tf.train.Saver()
with tf.Session as sess:
saver.save(sess,'my-model')#变量在sess中
'''
import tensorflow as tf
w1 = tf.Variable(tf.random_normal(shape=[2]), name='w1')
w2 = tf.Variable(tf.random_normal(shape=[5]), name='w2')
saver = tf.train.Saver()
sess = tf.Session()
sess.run(tf.global_variables_initializer())
saver.save(sess, 'my-model')
# my-model.data-00000-of-00001
# my-model.index
# my-model.meta
# checkpoint
如果只想保存其中一些变量,我们可以在创建tf.train.Saver实例的时候,给它传递一个我们想要保存的变量的list或者字典,若为空则保存所有变量。
import tensorflow as tf
w1 = tf.Variable(tf.random_normal(shape=[2]), name='w1')
w2 = tf.Variable(tf.random_normal(shape=[5]), name='w2')
saver = tf.train.Saver([w1,w2])
sess = tf.Session()
sess.run(tf.global_variables_initializer())
saver.save(sess, 'my-model')