Как визуализировать более одного ядра на слой в гистограммах, используя тензорную доску - PullRequest
Новая
       8

Как визуализировать более одного ядра на слой в гистограммах, используя тензорную доску

0 голосов
/ 11 декабря 2019

В настоящее время я использую Tensorflow 2.0 с простым CNN, я инициализирую первый слой с некоторыми фильтрами ручной работы, которые я хотел бы визуализировать в процессе обучения.

В части гистограммы тензорной доски я вижу толькопервое ядро ​​слоя, но я хотел бы увидеть все из них. Есть ли простой способ сделать это?

Заранее спасибо

1 Ответ

0 голосов
/ 16 декабря 2019

Создание маленькой функции, которая делает это при отображении обратного вызова в конце эпохи, - это способ, которым я решил это, не самый чистый, и было бы неплохо, если бы кто-то мог исправить это:) 

class DisplayCallback(tf.keras.callbacks.Callback):
def on_epoch_end(self, epoch, logs=None):
    variables_names = [v.name for v in model.trainable_variables]

    with file_writer_cm.as_default():
        for i in range(model.layers[0].get_weights()[0].shape[3]):
            tf.summary.histogram(variables_names[0].split('/')[0]+"/kernel_"+str(i), model.layers[0].get_weights()[0][:,:,:,i], step=epoch)
...