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について

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

、いくつかの便利な方法は、より良いを持っているために

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のコードをカスタマイズすることで可能です。