Google позволила визуализировать обучение нейросети в TensorFlow

Команда Google Brain сообщила о выпуске набора API для визуализации процесса обучения нейронных сетей. Частью данного релиза является обновление панели инструментов TensorBoard, которая идёт в комплекте с TensorFlow и отвечает за визуализацию, например, графиков изменения функции потерь во время обучения сети.

Инженеры команды Чи Женг и Джастин Танни прокомментировали релиз:

В отсутствие API добавление новых видов визуальной информации в TensorBoard было неимоверно трудной задачей для людей, не входящих в команду разработчиков TensorFlow. Таким образом, многие великолепные идеи для визуализации, предложенные сообществом, оставались лишь идеями без дальнейшего воплощения их в жизнь. Для того, чтобы исправить данное упущение, мы и выпустили в свободный доступ набор API, который позволит всем желающим добавлять свои настраиваемые плагины визуализации.

А как создавать плагины?

Разработчики предоставили пример написания плагинов для TensorFlow, чтобы пользователи как можно быстрее освоили основы создания плагинов и приступили к их реализации. Самым показательным примером того, какую свободу разработчики получили благодаря TensorBoard API, является Beholder. Этот плагин показывает видеодемонстрацию обучения модели в режиме онлайн, меняя значения градиентов и свёрточных фильтров.

Полный список уже доступных плагинов TensorBoard для таких вещей можно найти на Github.