当前位置: 移动技术网 > IT编程>脚本编程>Python > Pytorch中使用tensorboard可视化不显示的问题

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

2020年07月16日  | 移动技术网IT编程  | 我要评论

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

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

如对本文有疑问, 点击进行留言回复!!

相关文章:

验证码:
移动技术网