我正在尝试可视化我使用带有 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()

