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() with tf.Session() as sess: saver.restore(sess, "niu/save_net.ckpt") print("weights:", sess.run(W)) print("biases:", sess.run(b))
原文地址:https://www.cnblogs.com/yunyaniu/p/8353420.html
时间: 2024-10-01 06:25:21