Pytorch中TensorBoard及torchsummary的使用

最新推荐文章于 2025-05-15 17:22:53 发布

Mr_寒路

最新推荐文章于 2025-05-15 17:22:53 发布

阅读量3k

点赞数 8

分类专栏： Pytorch 文章标签：神经网络 pytorch

版权

本文介绍了如何在PyTorch中利用TensorBoard进行神经网络的可视化，包括创建SummaryWriter接口、记录标量、统计直方图、显示图像和查看模型图。此外，还讲解了torchsummary库用于查看网络层形状和参数的方法，并提供了启动TensorBoard的步骤。

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

1.TensorBoard神经网络可视化工具

TensorBoard是一个强大的可视化工具，在pytorch中有两种调用方法：

1.from tensorboardX import SummaryWriter
这种方法是在官方还不支持tensorboard时网上有大神写的

2.from torch.utils.tensorboard import SummaryWriter
这种方法是后来更新官方加入的

功能：创建接口

调用方法：writer = SummaryWriter("runs")

参数：

log_dir：event file输出文件夹

comment：不指定log_dir时，文件夹后缀

filename_suffix：event file文件名后缀

功能：记录标量add_scalars()

调用方法：writer.add_scalars("name",{"dic":val},epoch)

参数：

tag：图像的标签名

scalar_step：要记录的标量

global_step：轮次
</