Question

是否可以在Colaboratory中使用Tensorboard。在本地运行tensorboard会显示有关模型行为（如丢失等）的丰富信息。在使用Colaboratory（https://colab.research.google.com）时，可能会获得相同的信息。

Answer 1

您有两个选项可以使用其中一个python程序，它允许您隧道到托管您的python应用程序的机器实例。我测试了这个： https://github.com/taomanwai/tensorboardcolab

!pip install -U tensorboardcolab
from tensorboardcolab import *
import shutil

#clean out the directory
shutil.rmtree('./Graph', ignore_errors=True)
os.mkdir('./Graph')

tf.reset_default_graph()

#will start the tunneling and will print out a link:
tbc=TensorBoardColab()

#**here you construct your model**

sess = tf.Session()
output = sess.run(....)
sess.close()

train_writer = tbc.get_writer();
train_writer.add_graph(sess.graph)

train_writer.flush();
tbc.close()

另一个解决方案是压缩所有文件并将其下载到您的计算机上。

Answer 2

现在，您可以在Colab中使用Tensorboard，而无需使用ngrok和其他软件包：

import os
logs_base_dir = "tb_runs"
os.makedirs(logs_base_dir, exist_ok=True)
%load_ext tensorboard
%tensorboard --logdir {logs_base_dir}
# Now Tensorboard interface appear in this cell output

官方示例：https://www.tensorflow.org/tensorboard/get_started

如何在Colaboratory中使用Tensorboard

2 个答案: