首页 > 其他 > 详细

TF:利用TF的train.Saver载入曾经训练好的variables(W、b)以供预测新的数据

时间:2018-01-25 22:35:20      阅读:250      评论:0      收藏:0      [点我收藏+]
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))

  技术分享图片

 

TF:利用TF的train.Saver载入曾经训练好的variables(W、b)以供预测新的数据

原文:https://www.cnblogs.com/yunyaniu/p/8353420.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!