如何使用Keras的Tensorboard回调?

2023-12-30

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

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

如中所解释的keras.io https://keras.io/callbacks/#tensorboard。当我运行回调时我得到<keras.callbacks.TensorBoard at 0x7f9abb3898>,但我的文件夹“Graph”中没有收到任何文件。我使用这个回调的方式有问题吗?


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 
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

如何使用Keras的Tensorboard回调? 的相关文章

随机推荐