Tensorflow训练过程中查看某参数的值

有时候训练过程中想要打印某一可训练参数的值,查看训练效果,可以使用如下方法

import tensorflow as tf

with tf.name_scope('A'):
	var = tf.Varable(initial_value = tf.zeros([3,3]), name = 'var')
	
with tf.Session() as sess:
	varvar = sess.graph.get_tensor_by_name('A/var:0')
	print(sess.run(varvar))

如果不确定变量var的name,也可以用下面的方法获得

with tf.Session() as sess:
	for v in tf.trainable_variables():
    	print(v.name)
发布了42 篇原创文章 · 获赞 34 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/weixin_41024483/article/details/88543123