Tensorflowトレーニング中にパラメータの値を表示

2653 ワード

トレーニング中にトレーニング可能なパラメータの値を印刷し、トレーニング効果を表示するには、次の方法を使用します.
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)