问题来源
最近刚学习pytorch,看官方文档学习到了tensorboard可视化,但是照着代码写却不能得到图片显示
from torch.utils.tensorboard import SummaryWriter
# default `log_dir` is "runs" - we'll be more specific here
writer = SummaryWriter('runs/fashion_mnist_experiment_1') # 会创建一个log文件夹
dataiter = iter(trainloader)
images, labels = next(dataiter)
# create grid of images
img_grid = torchvision.utils.make_grid(images)
# show images
matplotlib_imshow(img_grid, True)
# write to tensorboard
writer.add_image('four_fashion_mnist_images', img_grid)
上面是官方文档代码,然后它说运行tensorboard --logdir=runs,再到浏览器输入localhost:6006即可在tensorboard界面端看到图片,然而我在cmd命令行窗口输入tensorboard --logdir=runs后再打开浏览器输入网址并没图片显示。
解决
首先找到你记录的tensorboard的log文件,代码中是writer = SummaryWriter(‘runs/fashion_mnist_experiment_1’),在我的电脑中找到全路径:D:\Jupyter_file\pytorch\GAN\runs,然后到cmd命令行输入cd /d D:\Jupyter_file\pytorch\GAN 切换到更具体的目录环境,具体如下:
然后在这个环境输入tensorboard --logdir=runs,再在浏览器输入localhost:6006打开tensorboard就可以看到图片显示啦!!!
效果如下:
PS:在同一文件夹的日志文件只会显示最开始运行的结果