tensorflow2.0 - WandbCallback 中的“save_graph”关键字是什么意思？

Question

我正在使用权重和偏差来跟踪我的深度学习模型。为了监视我使用的所有WandbCallback内容.fit。在WandbCallback 文档save_graph中有默认为 True的关键字。描述非常简短，我想知道保存的图表是什么以及它的用途是什么？保存图表是一项昂贵的操作吗？为什么需要它？（就像它补充了其他东西一样，比如保存最好的模型？）

标签： tensorflow2.0tf.keraswandb

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