monitoring - Keras - 在训练期间使用 TensorBoard 监控数量

Question

借助 Tensorflow，可以使用 tf.summary 监控训练期间的数量。

是否可以使用 Keras 执行相同的操作？您可以通过修改 https://github.com/fchollet/keras/blob/master/examples/variational_autoencoder.py 处的代码来包含一个示例吗？并监控 KL 损失(定义 at line 53 )

提前谢谢您!

Answer 1

您尝试过 TensorBoard 回调吗？ [1]

tensorboard = keras.callbacks.TensorBoard(log_dir='./logs',
                 histogram_freq=1, 
                 write_graph=True, 
                 write_images=False)
vae.fit(x_train,
        shuffle=True,
        epochs=epochs,
        batch_size=batch_size,
        validation_data=(x_test, x_test),
        callbacks=[tensorboard])

然后运行:

tensorboard --logdir=./logs

您可以编写回调的修改版本来处理您感兴趣的特定项目。

[1] https://keras.io/callbacks/#tensorboard