Saver保存提取

来源:互联网 发布:网络架构师年薪 编辑:程序博客网 时间:2024/06/05 18:44
# import tensorflow as tf## W = tf.Variable([[20,2,1],[212,12,1]],dtype=tf.float32,name="weights")# b = tf.Variable([[23,12,3]],dtype=tf.float32,name="biases")## saver = tf.train.Saver()## init = tf.global_variables_initializer()## with tf.Session() as sess:#     sess.run(init)#     save_path = saver.save(sess,"my_net/save_net.ckpt")#     print("Save to path:",save_path)import tensorflow as tfimport numpy as npW = tf.Variable(np.arange(6).reshape(2,3),dtype=tf.float32,name="weights")b = tf.Variable(np.arange(3).reshape(1,3),dtype=tf.float32,name="biases")saver = tf.train.Saver()init = tf.global_variables_initializer()with tf.Session() as sess:    sess.run(init)    saver.restore(sess,"my_net/save_net.ckpt")    print("weights",sess.run(W))    print("biases",sess.run(b))

原创粉丝点击