python-3.x - 在不使用 fit 方法中的回调的情况下在 Tensorboard 中显示 Keras 图形

Question

是否可以在不使用 fit 方法中的 tensorboard 回调的情况下在 Tensorboard 中显示 Keras 图形？

是否可以从 Keras 中提取图形并使用 tensorflow FileWriter 显示图形？tf.summary.FileWriter(logdir='logdir', graph=graph)

我想这样做是为了检查这部分图表的所有连接是否符合预期(这个模型是一个远未完成的更大模型的一部分)。

谢谢。

Answer 1

事实证明，从后端提取 Tensorflow 图并使用文件编写器非常简单。

import tensorflow as tf
# Used to get the graph
from tensorflow.python.keras import backend as K


tb_path = "logs/"

# Simple model to test the tensorboard plotting
model = SimpleModel(50, 20, 10).build_model()

# Get the sessions graph
graph = K.get_session().graph

# Display with the tensorflow file writer
writer = tf.summary.FileWriter(logdir=tb_path, graph=graph)