使用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的代码来实现。