2017-02-17 8 views
7

Nie mogę znaleźć sposobu dodania podsumowania wizualizacji wag dla sieci zdefiniowanych za pomocą tf.contrib.slim lub tf.contrib.layers.Korzystanie z podsumowania z tf slim lub tf layer

Na przykład, jeśli mam:

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

Jak mogę dodać wagi i stronniczości w podsumowaniu?

Edycja: Właśnie zobaczyłem, że mogę użyć tf.contrib.layers.summarize_collection. To prawdopodobnie działa na to, co chcę robić.

Odpowiedz

7

Oto rozwiązanie, które znalazłem po dokładniejszym zbadaniu, na wypadek gdyby ktoś miał ten sam problem niż ja.

Na tf.contrib.slim (wycofane) ciężary dodano w celu zbierania tf.GraphKeys.TRAINABLE_VARIABLES, które mogą być widoczne z:

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

Na tf.contrib.layers, istnieje kilka metod wygody

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

Aby lepiej kontrola, do której dodają zmienne, jest to możliwe poprzez dostosowanie kodu summarize_collection.

+0

'tf.contrib.layers.summarize_weights() # tf.GraphKeys.WEIGHTS tf.contrib.layers.summarize_biases() # tf.GraphKeys.BIASES' usunięto z modułów tensorflow. Czy możesz zaktualizować odpowiedź? – Yanagiba

+2

Domyślam się, że pytanie może zostać zaktualizowane za pomocą 'tf.contrib.layers.summarize_tensors()' – juliohm

Powiązane problemy