欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Pytorch中使用tensorboard可视化不显示的问题

程序员文章站 2023-11-01 13:54:04
Pytorch中使用tensorboard可视化不显示的问题问题来源解决问题来源最近刚学习pytorch,看官方文档学习到了tensorboard可视化,但是照着代码写却不能得到图片显示from torch.utils.tensorboard import SummaryWriter# default `log_dir` is "runs" - we'll be more specific herewriter = SummaryWriter('runs/fashion_mnist_experim...

Pytorch中使用tensorboard可视化不显示的问题

问题来源

最近刚学习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 切换到更具体的目录环境,具体如下:
Pytorch中使用tensorboard可视化不显示的问题
然后在这个环境输入tensorboard --logdir=runs,再在浏览器输入localhost:6006打开tensorboard就可以看到图片显示啦!!!
Pytorch中使用tensorboard可视化不显示的问题
效果如下:
Pytorch中使用tensorboard可视化不显示的问题
PS:在同一文件夹的日志文件只会显示最开始运行的结果

本文地址:https://blog.csdn.net/qq_41845478/article/details/107357117