tensorflow权值保存

#保存代码

import tensorflow as tf
W=tf.Variable([2,3],dtype=tf.float32,name="weight")  #W=[2,3]
b=tf.Variable([3],dtype=tf.float32,name="biases")

init=tf.initialize_all_variables()
saver=tf.train.Saver()
with tf.Session() as sess:
    sess.run(init)
    path=saver.save(sess,"path/weights.ckpt")
    print("path",path)

#恢复代码

W=tf.Variable(np.arange(2).reshape((1,2)),dtype=tf.float32,name="weight")
b=tf.Variable(np.arange(1).reshape((1,1)),dtype=tf.float32,name="biases")

saver=tf.train.Saver()
with tf.Session() as sess:

        saver.restore(sess,"path/weights.ckpt")
        print("w,b",sess.run(W),sess.run(b))

##注：权值保路径当前文件下

path/weights.ckpt