TensorFlow 的 Saving&Restoring

tf.train.Saver():

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)

# Add an op to initialize the variables.

init_op = tf.global_variables_initializer()

# Add ops to save and restore all the variables.

saver = tf.train.Saver()

# saver = tf.train.Saver({"v2": v2})

# Later, launch the model, initialize the variables, do some work, and save the

# variables to disk.

with tf.Session()as sess:

  sess.run(init_op)

 # Do some work with the model.

  inc_v1.op.run()

  dec_v2.op.run()

 # Save the variables to disk.

  save_path = saver.save(sess,"/tmp/model.ckpt")

# saver.restore(sess,"/tmp/model.ckpt")

 print("Model saved in file: %s" % save_path)