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

时间:2024-04-03 13:56:06

保存数据:

在代码中:

首先实例化一个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打开保存了该路劲即可

网页截图:

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