tensorboardX是基于tensorboard所做的一个用于pytorch数据可视化的一款工具。可以支持标量、图像、音频、文本、pytorch中搭建的网络结构等等。
安装：
代码语言：javascript复制pip install tensorboardX绘制标量简单实例：
代码语言：javascript复制from tensorboardX import SummaryWriter
writer = SummaryWriter("log")
for i in range(100):
writer.add_scalar("a", i, global_step=i)
writer.add_scalar("b", i**2, global_step=i)
writer.close()这时候会在当前目录生成一个log文件夹，在terminal命令行输入：
代码语言：javascript复制tensorboard --logdir log/会返回：
打开其中的链接：
可以看到这是我们创建的a和b的标量。
