テンソルフローを使ってサクッとネットワークを作ってみたところ、ちゃんとトレーニングできているようです。セッションを作成して確認したすべてのネットワーク操作に GPU を使用します。tf.Session(config=tf.ConfigProto(log_device_placement=True))
次のようにネットワークをセットアップしました。
train, image_guess = network(images, labels)
tf.summary.image('Guess', image_guess, max_outputs=3)
tf.summary.image('Input', images, max_outputs=3)
tf.summary.image('Target', tf.image.hsv_to_rgb(labels), max_outputs=3)
sess.run([
tf.local_variables_initializer(),
tf.global_variables_initializer(),
])
ネットワーク関数がグラフを設定し、トレーニング op と結果の画像テンソルをログ用に返します。
def network(images, labels):
with tf.variable_scope("NN") as scope:
...
return train_step, image_guess
問題は、テンソルボードでグラフを視覚化することです。何らかの理由で、次のような入力処理操作のグラフのみが表示されます。
ネットワーク全体を適切に表示するにはどうすればよいですか?