在Keras Functional API中注册回调



我正在从https://github.com/fchollet/keras/blob/master/examples/pretrained_word_embeddings.py上的预训练词嵌入模型示例中学习

我想使用Tensorboard可视化训练过程。Keras有一个用于Sequential API的Tensorboard回调,地址为https://keras.io/callbacks/#tensorboard,但没有用于示例中使用的Model API。

如何使用tensorboard可视化Model API网络?

回调在这两个api中都能很好地工作。
我目前在使用模型API构建的模型中使用Tensorboard回调,因此我可以确认。
在使用任何API定义模型并编译它之后,只需调用

tensorboard = keras.callbacks.TensorBoard()
model.fit(self, x, y, callbacks=[tensorboard])

相关内容

  • 没有找到相关文章

最新更新