跳到主要内容

使用 TensorBoard

为了更方便 TensorFlow 程序的理解、调试与优化,TensorFlow 团队发布了一套名为 TensorBoard 的可视化工具。可以用 TensorBoard 来展现 TensorFlow 图,绘制图像生成的定量指标图以及显示附加数据(如其中传递的图像)。

目前 OpenBayes 容器在创建 Task 或 Jupyter 工作空间时都已经支持了 TensorBoard。

以下的代码示例展示了如果在训练时指定使用 TensorBoard:

def create_model():  model = tf.keras.models.Sequential([  tf.keras.layers.Dense(512, activation=tf.nn.relu, input_shape=(img_rows * img_cols,)),  tf.keras.layers.Dropout(0.2),  tf.keras.layers.Dense(num_classes, activation=tf.nn.softmax)])
  model.compile(optimizer='adam',              loss='sparse_categorical_crossentropy',              metrics=['accuracy'])
  return model
model = create_model()model.summary()
log_path = './tf_dir'
# 设置 TensorBoard 路径到 tf_dirtb_callback = tf.keras.callbacks.TensorBoard(log_dir=log_path)
model.fit(x_train, y_train,          epochs=epochs,          verbose=1,          callbacks=[tb_callback])

完整的代码见 MNist Example

注意

如下图所示,我们默认指定 TensorBoard 数据获取的目录为 /output 下的 tf_dir 文件夹。因此如果您需要展示 TensorBoard 数据,请务必将 TensorBoardlog_dir 设置为该目录。

对于运行成功的容器,其页面上会出现 "打开 TensorBoard" 的按钮,打开就可以看到正在运行的 TensorBoard 了。

容器只有在运行的状态下才能查看 TensorBoard,如果您想在容器停止后依然查看 TensorBoard,请将 "容器输出" 下载到本地,并通过在本地安装 TensorBoard 来查看其中的内容:

tensorboard --logdir=<TensorBoard 数据目录>

然后在浏览器打开 http://localhost:6006/ 即可。