【Pytorch】torch.utils.tensorboard

rejudge

已于 2023-05-06 14:19:27 修改

阅读量283

点赞数

CC 4.0 BY-SA版权

分类专栏： Python 文章标签： pytorch 深度学习人工智能

于 2023-04-22 18:34:09 首次发布

本文链接：https://blog.csdn.net/qq_45249685/article/details/130308437

Python 专栏收录该内容

43 篇文章

订阅专栏

本文介绍了如何使用Tensorboard进行日志可视化，包括通过SummaryWriter创建logs目录，用add_image展示图片，以及利用add_graph来查看模型结构。这些工具对于深度学习模型的训练监控非常有帮助。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 使用方法

写好代码，生成文件夹logs
到logs所在目录（当前目录）下，打开终端运行 tensorboard --logdir=logs
根据提示打开页面 http://localhost:6006/

2. 看图片 SummaryWriter().add_image()

from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms

writer = SummaryWriter('logs')
img_tensor = torch.randn((3, 3200, 1800))
writer.add_image('test', img_tensor)
writer.close()

在这里插入图片描述

2. 看模型 SummaryWriter().add_graph()

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter('logs')
writer.add_graph(MyModule(), torch.randn(2, 3, 256, 256))
writer.close()

在这里插入图片描述