tf 슬림 또는 tf 레이어의 요약 사용

Question

tf.contrib.slim 또는 tf.contrib.layers으로 정의 된 네트워크의 가중치를 시각화하기 위해 요약을 추가하는 방법을 찾을 수 없습니다. 예를 들어

, 내가있는 경우 :

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

어떻게 가중치 및 바이어스 요약에 추가 할 수 있습니까?

편집 : 방금 tf.contrib.layers.summarize_collection을 사용할 수있는 것으로 보았습니다. 아마 내가하고 싶은 일에 효과가있을 것입니다.

Answer 1

누군가가 나보다 똑같은 문제가있는 경우에 대비하여 더 깊은 조사를 한 후에 발견 한 해결책입니다.

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() 

이 더 나은을 가지고있다 : tf.contrib.slim를 들어

는 (사용되지 않음) 가중치는로 시각화 할 수 있습니다 컬렉션 tf.GraphKeys.TRAINABLE_VARIABLES에 추가 어떤 변수를 추가할지 제어하려면 summarize_collection의 코드를 맞춤 설정하면됩니다.