我正在尝试可视化我使用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()
发布于 2019-11-21 09:42:54
我也有同样的问题。查看这个线程:https://github.com/pytorch/pytorch/issues/24157
TLDR:将PyTorch更新为PyTorch夜间,问题应该得到解决.https://pytorch.org/get-started/locally/
https://datascience.stackexchange.com/questions/57952
复制