一、什么是Tensorboard
Tensorboard原本是Google TensorFlow的可视化工具,可以用于记录训练数据、评估数据、网络结构、图像等,并且可以在web上展示,对于观察神经网络的过程非常有帮助。PyTorch也推出了自己的可视化工具,一个是 tensorboardX 包,一个是 torch.utils.tensorboard,二者的使用相差不大,这里介绍后者
————————————————
开始实操:
1.首先生成文件夹logs
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter("logs")
for i in range(100):
writer.add_scalar("y=x",i,i)
writer.close()
终端输入 显示图像
tensorboard --logdir=logs
更改输出端口 默认为6006
tensorboard --logdir logs --port=6007
效果图
NOTE:最好每一个图像对应一个文件夹!!就是要重新创建一个SummaryWriter对象。