带有pytorch的张量板不显示图表

数据挖掘 深度学习 可视化 火炬 卷积神经网络
2022-03-07 20:18:48

我正在尝试可视化我使用带有 Pytorch 的 Tensorboard 创建的模型,但是在运行 tensorboard 并转到图形选项卡时没有显示任何内容,我添加了我的代码以供参考,我还为所有依赖项添加了我的 conda env 的屏幕截图

from torch.utils.tensorboard import SummaryWriter 

writer = SummaryWriter()
images, labels = next(iter(trainloader))
grid = torchvision.utils.make_grid(images)

writer.add_image('images', grid)
writer.add_graph(model=model, input_to_model=images, verbose=True)
writer.close()

Conda env 运行代码

张量板的屏幕截图试图显示我的模型图

1个回答

我有同样的问题。看看这个线程: https ://github.com/pytorch/pytorch/issues/24157

TLDR:将 PyTorch 更新为 PyTorch-nightly,问题应该得到解决。 https://pytorch.org/get-started/locally/