模型的保存
import tensorflow as tf
v1 = tf.Variable(tf.random_normal([1,2]),name='v1')
v2 = tf.Variable(tf.random_normal([2,3]),name='v2')
init = tf.global_variables_initializer()
saver = tf.train.Saver()
with tf.Session() as sess:
sess.run(init)
print('v1:', sess.run(v1))
print('v2:', sess.run(v2) )
saver_path = saver.save(sess, 'saver/model.ckpt')
print('model saved in file :', saver_path)
模型的读取
import tensorflow as tf
v1 = tf.Variable(tf.random_normal([1,2]),name='v1')
v2 = tf.Variable(tf.random_normal([2,3]),name='v2')
saver = tf.train.Saver()
with tf.Session() as sess:
saver.restore(sess, 'saver/model.ckpt')
print('v1:', sess.run(v1))
print('v2:', sess.run(v2))
print('mdoel restored')