tensorboard的使用（一）：add_scalar()的使用

add_scalar()的使用，常用来绘制train/val loss

1.引入包

from torch.utils.tensorboard import SummaryWriter

2.#SummaryWriter()：自动生成文件夹名称"logs"

writer = SummaryWriter("logs")

3.调用add_scalar()函数自定义一个表达式.以y=x为例

add_scalar(①,②,③)

①字符串类型 “自定义表达式名 y=x”

②代表的是 y

③代表的是X

【例】画出y=2x ,x照常写，y写成2x就可以

writer.add_scalar("y=2x", 2*i, i)

for i in range(100):
    writer.add_scalar("y=x", i, i)

4.关闭write

writer.close()

【----------------运行程序-----------------】

在Terminal上输入命令 (logdir=事件文件所在的文件夹名)

tensorboard --logdir=logs --port=6007