# 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 tf
import numpy as np
W = 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))
Saver保存提取
最新推荐文章于 2021-11-15 20:47:39 发布