1、
import tensorflow as tf
import numpy as np
# save to file
W = tf.Variable([[1,2,3],[4,5,6]],dtype = tf.float32,name='weight')
b = tf.Variable([[1,2,3]],dtype = tf.float32,name='biases')
init = tf.initialize_all_variables()
saver = tf.train.Saver()
with tf.Session() as sess:
sess.run(init)
save_path = saver.save(sess,"my_net/save_net.ckpt")
print ("save to path:",save_path)
说明:保存模型
2、
import tensorflow as tf
import numpy as np
W = tf.Variable(np.arange(6).reshape((2,3)),dtype = tf.float32,name='weight')
b = tf.Variable(np.arange(3).reshape((1,3)),dtype = tf.float32,name='biases')
saver = tf.train.Saver()
with tf.Session() as sess:
saver.restore(sess,"my_net/save_net.ckpt")
print ("weights:",sess.run(W))
print ("biases:",sess.run(b))
说明:加载模型