Команда Google Brain сообщила о выпуске набора API для визуализации процесса обучения нейронных сетей. Частью данного релиза является обновление панели инструментов TensorBoard, которая идёт в комплекте с TensorFlow и отвечает за визуализацию, например, графиков изменения функции потерь во время обучения сети.
Инженеры команды Чи Женг и Джастин Танни прокомментировали релиз:
В отсутствие API добавление новых видов визуальной информации в TensorBoard было неимоверно трудной задачей для людей, не входящих в команду разработчиков TensorFlow. Таким образом, многие великолепные идеи для визуализации, предложенные сообществом, оставались лишь идеями без дальнейшего воплощения их в жизнь. Для того, чтобы исправить данное упущение, мы и выпустили в свободный доступ набор API, который позволит всем желающим добавлять свои настраиваемые плагины визуализации.
А как создавать плагины?
Разработчики предоставили пример написания плагинов для TensorFlow, чтобы пользователи как можно быстрее освоили основы создания плагинов и приступили к их реализации. Самым показательным примером того, какую свободу разработчики получили благодаря TensorBoard API, является Beholder. Этот плагин показывает видеодемонстрацию обучения модели в режиме онлайн, меняя значения градиентов и свёрточных фильтров.
Полный список уже доступных плагинов TensorBoard для таких вещей можно найти на Github.
Источник: блог Google Research
