tensorboardX可视化

tensorboardX是基于tensorboard所做的一个用于pytorch数据可视化的一款工具。可以支持标量、图像、音频、文本、pytorch中搭建的网络结构等等。

安装：

pip install tensorboardX

绘制标量简单实例：

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命令行输入：

tensorboard --logdir log/

会返回：

打开其中的链接：

可以看到这是我们创建的a和b的标量。