tensorflow

读取
sess.run(tf.get_default_graph().get_tensor_by_name('test:0')

saver.restore(sess,'/ckpt/rnn.ckpt')
graph = tf.get_default_graph()
pred_y = tf.get_collection('pred_y')[0]
X = graph.get_operation_by_name('X').outputs[0]    X为placeholder  所以使用 operation_by_name
Y = graph.get_operation_by_name('Y').outputs[0]



一定要注意 saver=tf.train.Saver() 的位置 在所有变量定义后! 大坑

  

posted @ 2018-08-08 13:42  _陈昱先  阅读(191)  评论(0)    收藏  举报