使用tf slim或tf图层的摘要

Question

我无法找到如何添加摘要来可视化已使用tf.contrib.slim或定义的网络的权重tf.contrib.layers.

例如,如果我有:

net = slim.conv2d(net, ...)

如何将权重和偏差添加到摘要中？

编辑:我刚看到我可以使用tf.contrib.layers.summarize_collection.这可能适用于我想做的事情.

Answer 1

这是我经过深入调查后发现的解决方案,以防有人遇到与我相同的问题.

对于tf.contrib.slim(已弃用)权重将添加到集合中tf.GraphKeys.TRAINABLE_VARIABLES,可以通过以下方式显示:

tf.contrib.layers.summarize_collection(tf.GraphKeys.TRAINABLE_VARIABLES)

对于tf.contrib.layers,有一些便利方法

tf.contrib.layers.summarize_weights()  # tf.GraphKeys.WEIGHTS
tf.contrib.layers.summarize_biases()  # tf.GraphKeys.BIASES
# For tf >= 1.3:
tf.contrib.layers.summarize_tensors()

为了更好地控制添加哪些变量,可以通过自定义summarize_collection的代码来实现.