Pytorch 可视化之tensorboardX

最新推荐文章于 2024-10-29 09:04:32 发布

原创最新推荐文章于 2024-10-29 09:04:32 发布 · 872 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#tensorboardX #pytorch #python #可视化

心得同时被 3 个专栏收录

69 篇文章

订阅专栏

python

20 篇文章

订阅专栏

pytorch

13 篇文章

订阅专栏

本文详细介绍如何使用TensorBoardX进行模型训练过程的可视化监控，包括安装配置、基本用法及示例代码，帮助读者快速上手。

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

使用tensorboardX需要安装tensorflow和tensorboardX，安装语句如下：

pip install tensorboardX
pip install tensorflow

其核心的语句为三句：

from tensorboardX import SummaryWriter

 writer = SummaryWriter('./log/')  # 设置保存的log文件夹
 writer.add_scalar('name', y, x) # 设置图的名字，和y、x的值，注意这里不是x,y，而是y,x
 writer.close() # 关闭输入

接下来我们来举个栗子吧～

from tensorboardX import SummaryWriter

if __name__ == '__main__':
    # x = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
    writer = SummaryWriter('./log/')
    for i in range(1, 10):
        writer.add_scalar('test', i, i)
    writer.close()

在这里插入图片描述