Что означает ключевое слово save_graph в WandbCallback?

Question

Я использую веса и смещения, чтобы отслеживать свои модели глубокого обучения. Чтобы контролировать все, я использую WandbCallback в .fit. В документации WandbCallback есть ключевое слово save_graphкоторый по умолчанию имеет значение True. Описание очень краткое, и мне было интересно, что это за сохраненный график и для чего он нужен? Сохранение графика - дорогостоящая операция? Для чего это нужно? (например, дополняет ли он что-то еще, например, сохраняет лучшую модель?)

Answer 1

Это используется для создания журнала wandb.Graph модели. Этот класс обычно используется для сохранения и отображения моделей нейронных сетей. Он представляет граф как массив узлов и ребер. Узлы могут иметь метки, которые можно визуализировать с помощью wandb. Вот пример графика, который он создает: https://wandb.ai/l2k2/keras-examples/runs/ieqy2e9h/model

Вот код, который делает это в обратном вызове. https://github.com/wandb/client/blob/1609f82c84e2244ed8fe62c746099d2094bd746a/wandb/integration/keras/keras.py#L552