tensorflow - GPU Ops 未显示在张量板图上

Question

我已经使用 tensorflow 建立了一个快速网络，它似乎正在正确训练。它使用 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(),
])

网络函数设置图形并返回训练操作和生成的图像张量以进行记录。

def network(images, labels):
    with tf.variable_scope("NN") as scope:
        ...
        return train_step, image_guess

麻烦在于在 tensorboard 中可视化图形。出于某种原因，它只显示了我的输入处理操作的图表，如下所示：

如何让它正确显示我的整个网络？