我可以使用Google Colab来使用TensorBoard吗?

109

在Google Colab训练TensorFlow模型时,有没有办法使用TensorBoard?


4
官方文档链接:https://colab.research.google.com/github/tensorflow/tensorboard/blob/master/docs/tensorboard_in_notebooks.ipynb - EFreak
现在,您可以简单地使用TensorBoard魔法。%tensorboard --logdir your_path - Hissaan Ali
22个回答

0

目前我发现的最简单和最容易的方法是:

使用wget获取setup_google_colab.py文件

!wget https://raw.githubusercontent.com/hse-aml/intro-to- dl/master/setup_google_colab.py -O setup_google_colab.py
import setup_google_colab

要在后台运行tensorboard,请暴露端口并单击链接。
我假设您已经在摘要中添加了适当的价值,并合并了所有摘要。

import os
os.system("tensorboard --logdir=./logs --host 0.0.0.0 --port 6006 &")
setup_google_colab.expose_port_on_colab(6006)

运行以上语句后,你会看到一个链接提示:

Open https://a1b2c34d5.ngrok.io to access your 6006 port

请参考以下 Git 获取更多帮助:
https://github.com/MUmarAmanat/MLWithTensorflow/blob/master/colab_tensorboard.ipynb

看看其他答案,上面的选项看起来更好,因为它们不需要手动步骤。 - lucid_dreamer
@lucid_dreamer,毫无疑问以上答案有更好的解释、步骤和易懂性,但如果我们知道其他答案,让其他人知道所有技巧也是很好的,也许某些技巧对某些人有帮助。这完全取决于你的选择。 - Muhammad Umar Amanat
我遇到了AttributeError: module 'setup_google_colab' has no attribute 'expose_port_on_colab'的错误。 - Biranchi
@Biranchi expose_port_on_colab函数在setup_google_colab.py中,您可以访问此链接以供参考https://github.com/hse-aml/intro-to-dl/blob/master/setup_google_colab.py#L105 - Muhammad Umar Amanat

-1

你可以试试这个,对我来说很有效

%load_ext tensorboard
import datetime
logdir = os.path.join("logs", datetime.datetime.now().strftime("%Y%m%d-%H%M%S"))
tensorboard_callback = tf.keras.callbacks.TensorBoard(logdir, histogram_freq=1)

 model.fit(x=x_train, 
        y=y_train, 
        epochs=5, 
        validation_data=(x_test, y_test), 
        callbacks=[tensorboard_callback])

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接