Tensorflow查看模型参数

2018-12-18  本文已影响0人  Fall_Q

既载入模型结构(Graph),也载入参数值

import tensorflow as tf

# importing graph

saver = tf.train.import_meta_graph("/home/jagger/workspace/tmp/model.ckpt.meta")

with tf.Session() as sess:

# loading variable value to sess

    saver.restore(sess, "/home/jagger/workspace/tmp/model.ckpt")

    result =  tf.get_default_graph().get_tensor_by_name("add:0")   

    print(sess.run(result)) 
上一篇 下一篇

猜你喜欢

热点阅读