什么是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.
如果生成的标签是相同的,再次绘制时就会出现乱图。就是会增加变化过程。建议就是删除所有的日志文件再次运行或者更改标签。