帶Keras和fit_generator的TensorBoard分佈和直方圖

Question

我使用Keras使用fit_generator函數來訓練CNN。

它似乎是known issue，TensorBoard在此設置中不顯示直方圖和分佈。

有沒有人想出辦法讓它工作呢？

Answer 1

有沒有簡單的方法來插入一行代碼，你必須手工編寫摘要。

好消息是，它並不難，你可以使用TensorBoard回調代碼在Keras作爲參考：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