在 Colab 上运行 TensorBoard

Posted 2023-03-27

【中文标题】在 Colab 上运行 TensorBoard

【英文标题】：Run TensorBoard on Colab

【发布时间】：2021-11-04 04:13:59

【问题描述】：

我有以下代码

!pip install timm

import timm
from torchsummary import summary
import torchvision.models as models
import torch 
import os
from torch.utils.tensorboard import SummaryWriter

%load_ext tensorboard
#model = models.inception_v3(pretrained=False)

model = timm.create_model('inception_v4', pretrained=False)
#model.cuda()

summary(model, (3, 440, 440))
writer=SummaryWriter('/content/logsdir')

%tensorboard --logdir /content/logsdir

代码运行时，出现 No graph definition files were found 错误。有什么问题？

【问题讨论】：

你能提供完整错误回溯吗？

你能在运行后在一个单元格中运行!ls /content/ 和!ls /content/logsdir 吗？我还建议尝试 wandb.ai 而不是 tensorboard 以获得更好的实验跟踪

@tnfru 我试过了，但还是一样。运行上述代码后，我运行!ls /content/ and !ls /content/logsdir。

@Ivan 你能在 Colab 中运行代码吗？这段代码不需要任何东西，可以直接在 Colab 中运行

我确实运行了它，它正确地启动了嵌入式 Tensorboard 窗口。这可能意味着问题出在你身上......

【参考方案1】：

使用内置魔法命令 https://ipython.readthedocs.io/en/stable/interactive/magics.html

阅读参考文档https://colab.research.google.com/github/tensorflow/tensorboard/blob/master/docs/tensorboard_in_notebooks.ipynb