TensorFlow模型保存和恢复简单的例子

#%%保存
import tensorflow as tf
v1 = tf.get_variable("v1",shape=[3],initializer = tf.zeros_initializer)
v2 = tf.get_variable("v2",shape=[5],initializer = tf.zeros_initializer)

inc_v1 = v1.assign(v1+1)
dec_v2 = v2.assign(v2-1)

init_op = tf.global_variables_initializer()

saver = tf.train.Saver()

with tf.Session() as sess:
    sess.run(init_op)
    inc_v1.op.run()
    dec_v2.op.run()
    save_path = saver.save(sess,"D:/tmp/model.ckpt")
    print("model saved in file:%s" % save_path)

sess.close()
#%%恢复
import tensorflow as tf
v1 = tf.get_variable("v1",shape=[3])
v2 = tf.get_variable("v2",shape=[5])
saver = tf.train.Saver()
with tf.Session() as sess:
    saver.restore(sess,"D:/tmp/model")
    print("v1:%s"%v1.eval())
    print("v2:%s"%v2.eval())