tensorflow 如何在tensorboard中可视化tf.compat.v1静态图形？

dldeef67 于 2023-01-17 发布在其他

关注(0)|答案(1)|浏览(140)

对于一个给定的图，我们如何使用tf.compat.v1的tensorboard来可视化这个图呢？在搜索完所有地方之后，在这里分享这个。大多数文档都解释了tf.keras，而不是tf.compat.v1的静态图

1条答案

首先导出图表到一个日志目录，tensorboard可以使用！

import tensorflow as tf

# Get the default graph
graph = tf.compat.v1.get_default_graph()
writer = tf.compat.v1.summary.FileWriter("logs", graph)

之后，只需在指定的目录下打开tensorboard（这里是logs/）
python -m tensorboard.main --logdir logs/