如何使用tensorboardX保存以及可视化训练过程数据

保存数据：

在代码中：

首先实例化一个SummaryWriter

import tensorboardX as tb

writer = tb.SummaryWriter(log_dir=tb_path)

然后将需要保存的数据以键-值的方式保存在writer中：

writer.add_scalar(key, value, step)

其中step表示了保存的值来自第几次迭代，比如我第三次迭代的train_loss=0.3，那么key就为train_loss，value=0.3， step=3

 

保存完数据后，可视化数据：

在终端输入命令：

tensorboard --logdir=logs/tblog --port=6006

logs/tblog为我们之前保存的数据，port代表我们通过哪个端口输出

输入命令后，按照提示，打开相应的网页即可（不可以用360浏览器）

如果需要在一张图上展现多条曲线，只需要将不同曲线的数据以相同的名称保存在同一路径下，在终端输入命令时--logdir打开保存了该路劲即可

网页截图：