保存数据:
在代码中:
首先实例化一个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打开保存了该路劲即可
网页截图: