我在TF中建立CNN模型。
我保存了一些变量
wc1 = tf.Variable(tf.random_normal([5, 5, 1, 32]), name='wc1')
wc2 = tf.Variable(tf.random_normal([5, 5, 32, 64]), name='wc2')
通过
^{pr2}$当我在另一个会话中恢复保存的模型并使用tf.打印(),无法打印。下面的代码用于恢复模型
sess = tf.Session()
saver = tf.train.import_meta_graph("./cnn_model.meta")
saver.restore(sess, './cnn_model')
wc1 = tf.get_default_graph().get_tensor_by_name("wc1:0")
wc2 = tf.get_default_graph().get_tensor_by_name("wc2:0")
while some_step:
sess.run(optimizer, feed_dict={x: batch_x, y: batch_y})
wc1 = tf.Print(wc1, [wc1], 'WC1 is: ')
如何打印/获取保存模型的张量值?
只需执行
sess.run(wc1)
,就可以获得模型的值。下面是我的代码示例:相关问题 更多 >
编程相关推荐