显示tensorflow图中的变量

显示tensorflow图中的变量

 

1、可训练变量

 tf.trainable_variables()

[var.name for var in tf.trainable_variables()]

#print

['conv1/kernel:0',
 'conv1/bias:0',
 'conv2/kernel:0',
 'conv2/bias:0',
 'fc1/kernel:0',
 'fc1/bias:0',
 'output/kernel:0',
 'output/bias:0']

2、全部的变量

tf.global_variables()

[var.name for var in tf.global_variables()]

3.收集变量的信息，用于存储预训练的值

{var.name:var.value for var in tf.trainable_variables()}

或者：

{var.name:var.value for var in tf.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES)}