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:

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

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

path/weights.ckpt

文章来源: tensorflow权值保存