TensorboardX简易安装和使用教程

1.安装

首先安装很简单，不需要什么tensorflow之类的直接安装。

pip install tensorboardX

2.程序中使用

废话不多说，说白了这个tensorboard 就是传入数据画图。

from tensorboardX import SummaryWriter
writer = SummaryWriter('runs/scalar_example')

runs/scalar_example 这个是它生成文件的存放地址。

接下来就是重点：

    writer.add_scalar('GLoss', Gloss/sum, global_step=epoch)
    writer.add_scalar('DLoss', Dloss/sum, global_step=epoch)
    writer.add_scalar('RMSE', Rmse/sum, global_step=epoch)
    writer.add_scalar('CC', CC/sum, global_step=epoch)

这个’GLoss’是图表的名字，这个图表的Y坐标：Gloss/sum X坐标epoch

比如说在我的程序里。

最后画出来的图表：

3.查看

其实你在程序运行时，它会自动生成这个文件

这个时候你要怎么打开查看这个tensorboard的网页呢？


复制这串地址，其实就是它生成的文件的地址。我喜欢这么来。

去你安装了tensorboard的那个环境，然后运行一下代码：

tensorboard --logdir=你复制的地址

如图

OK

打开这个网址http://localhost:6006/

出现了！！

so easy!