tensrflow基础学习

tensorflow基础学习

保存与恢复模型及变量

保存

#Create some variables.
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()
 #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")
  print("Model saved in path: %s" % save_path)

恢复

tf.reset_default_graph()
#Create some variables.
v1 = tf.get_variable("v1", shape=[3])
v2 = tf.get_variable("v2", shape=[5])

#Add ops to save and restore all the variables.
saver = tf.train.Saver()

#Later, launch the model, use the saver to restore variables from disk, and
#do some work with the model.
with tf.Session() as sess:
  #Restore variables from disk.
  saver.restore(sess, "/tmp/model.ckpt")
  print("Model restored.")
  #Check the values of the variables
  print("v1 : %s" % v1.eval())
  print("v2 : %s" % v2.eval())

检查某个检查点中的变量：

将模型中的tensor打印出来

from tensorflow.python.tools import inspect_checkpoint as chkp
chkp.print_tensors_in_checkpoint_file("/tmp/model.ckpt", tensor_name='', all_tensors=True)
chkp.print_tensors_in_checkpoint_file("/tmp/model.ckpt", tensor_name='v1', all_tensors=False)