在Tensorboard中显示Keras图而不在fit方法中使用回调

Question

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

是否可以从Tensorflow FileWriter中从Keras提取图形并显示图形？ 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)