pytorch - 在 Colab 上运行 TensorBoard

Question

我有以下代码

!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 found 错误。有什么问题？

标签： pytorchtensorboard

Answer 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阅读参考文档