Обратный вызов Keras в Tensorboard имеет аргумент write_images и histogram_freq, который позволяет сохранять веса и активации в Tensorboard для визуализации во время тренировки.
Проблема в том, что при этом сохраняется информация для каждого слоя и делает Tensorboard очень грязным, особенно если я записал другие изображения в Tensorboard. Это можно увидеть на изображениях ниже:
Большая часть этой зарегистрированной информации является избыточной. Есть ли способ сделать визуализацию распределения веса и активации более организованной? Есть ли способ визуализировать только определенные слои?