在 Colab上使用Tensorboard

TensorBoard在Colab Google工作

安装

!pip install tensorboardcolab

在Colab Google Jupyter中，为了自动安装并确保使用最新版本的TensorBoardColab，请在Jupyter单元的第一行添加“！pip install -U tensorboardcolab

要求

Tensorflow
TensorBoard
npm

初始化

from tensorboardcolab import TensorBoardColab
tbc=TensorBoardColab()

初始化后，TensorBoard链接将显示在Colab Google Juyter输出中

PS：如果初始化失败并继续重试，请将startup_waiting_time增加大于8秒，如下所示

tbc=TensorBoardColab(startup_waiting_time=30)

添加到Keras回调

model.fit(x,y,epochs=100000,callbacks=[TensorBoardColabCallback(tbc)])

将图片保存到TensorBoard

tbc.save_image(title="test_title", image=image)

将值保存到TensorBoard图表

tbc.save_value("graph_name", "line_name", epoch, value)
.
.
.
tbc.flush_line(line_name)
tbc.close()