Tensorboard的使用

什么是Tensorboard

在训练过程中我们能希望看到训练的过程怎么样，这时候就可以使用这个模块来实现。他可以显示函数之间的变化过程。

Tensorboard的使用

Summarywriter类

这个类的作用是创建训练日志文件。
Summarywriter（“文件地址”）就可以在目录下生成训练的日志，这个日志可以用来展示成可视化的board
通常使用的是add_scalar()生成绘制的日志。
包含三个参数，tag，y轴,x轴

from torch.utils.tensorboard import SummaryWriterwriter = SummaryWriter('logs')#比如绘制一个y=x*x的训练过程for x in range(10):writer.add_scalar("y=x*x",x*x,x)
writer.close()

运行之后就会在这里面生成一个logs的日志文件夹和内容

绘制board

运行代码后会生成日志文件，在命令行使用命令tensorboard --logdir=目录 --port=端口端口默认是6006.

如果生成的标签是相同的，再次绘制时就会出现乱图。就是会增加变化过程。建议就是删除所有的日志文件再次运行或者更改标签。