Kerasとfit_generatorを使用したTensorBoardの分布とヒストグラム

Question

私は、fit_generator関数を使用してCNNを訓練するためにKerasを使用しています。

TensorBoardはこの設定ではヒストグラムと分布を表示していないと思われるknown issueと思われます。

誰もがとにかく動作させる方法を見つけましたか？

Answer 1

1行のコードで簡単にプラグインする方法はありません。手作業でサマリーを書き込む必要があります。

グッドニュースは、それは難しいことではありません、あなたが参照としてKerasにTensorBoardコールバックコードを使用することができるということです。https://github.com/fchollet/keras/blob/master/keras/callbacks.py#L537

基本的には、例えば関数を書きますwrite_summaries(model)とあなたがtensorboardに表示するデータを記録するために、あなたのwrite_summaries(model)機能使用インサイド

tf.summary、histogram_summaryおよびその他の要約機能（例えば、ちょうどあなたのfit_generator()後に）あなたの要約を書きたい時はいつでもそれを呼び出します。 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

あなたが公式のチュートリアルを確認してください正確にどのように、わからない場合