keras - 如何使用Keras的Tensorboard回调？

Question

我用 Keras 构建了一个神经网络。我会通过 Tensorboard 可视化其数据，因此我使用了:

keras.callbacks.TensorBoard(log_dir='/Graph', histogram_freq=0,
                            write_graph=True, write_images=True)

如 keras.io 中所述。当我运行回调时，我得到 <keras.callbacks.TensorBoard at 0x7f9abb3898> ，但我的文件夹“Graph”中没有收到任何文件。我使用这个回调的方式有问题吗？

Answer 1

keras.callbacks.TensorBoard(log_dir='./Graph', histogram_freq=0,  
          write_graph=True, write_images=True)

此行创建一个 Callback Tensorboard 对象，您应该捕获该对象并将其提供给模型的 fit 函数。

tbCallBack = keras.callbacks.TensorBoard(log_dir='./Graph', histogram_freq=0, write_graph=True, write_images=True)
...
model.fit(...inputs and parameters..., callbacks=[tbCallBack])

这样你就可以将回调对象传递给函数。它将在训练期间运行，并输出可与张量板一起使用的文件。

如果您想可视化训练期间创建的文件，请在终端中运行

tensorboard --logdir path_to_current_dir/Graph