pytorch-tensorboard的基本用法一——add_scalar()

pytorch-tensorboard的基本用法一——add_scalar

1、tensorboard简介

主要是用来可视化的工具，可以可视化神经网络的训练过程，便于调参等等
具体如下所示：

2、代码实现

使用add_scalar描点连线，scalar_value横轴，global_step纵轴

from torch.utils.tensorboard import SummaryWriter

if __name__ == '__main__':
    writer = SummaryWriter("logs")
    for i in range(100):
        writer.add_scalar("y = 2x", 2 * i, i)

writer.close()

然后在terminal中输入

tensorboard --logdir=logs

默认端口6006

更改端口

tensorboard --logdir=logs --port=6007

点击url, 即可显示可视化的结果

3、注意

(1) 两个不同的文件的tag相同，结果会叠加，出现拟合等
(2) 可以删除之前生成的文件得到想要的结果