Jon*_*oth 18 python keras tensorflow tensorboard
我正在使用Keras使用fit_generator函数训练CNN.
似乎是TensorBoard在此设置中未显示直方图和分布的已知问题.
有没有人找到一种让它工作的方法？
没有简单的方法可以用一行代码插入它,你必须手动编写摘要.
好消息是它并不困难,您可以在Keras中使用TensorBoard回调代码作为参考:https://github.com/fchollet/keras/blob/master/keras/callbacks.py#L537
基本上,编写一个函数,例如write_summaries(model),每当你想要编写摘要时调用它(例如,在你的后面fit_generator())
在您的write_summaries(model)函数内部使用tf.summary,histogram_summary以及其他汇总函数来记录您想要在tensorboard上看到的数据.
如果您 不确切知道如何,请查看官方教程:https://www.tensorflow.org/get_started/summaries_and_tensorboard以及带有摘要的MNIST的一个很好的例子:https://github.com/tensorflow/tensorflow/blob/master /tensorflow/examples/tutorials/mnist/mnist_with_summaries.py