tensorboard_SummaryWriter_event定义及使用示例

一、导包并创建SummaryWriter

使用SummaryWriter创建events：

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter()
writer = SummaryWriter("test0")
writer = SummaryWriter(comment="user")

SummaryWriter()：自动生成文件夹名称"runs"，以及随机的events名称
SummaryWriter(“test0”)：自动生成文件夹名称"test0"，以及随机的events名称
SummaryWriter(comment=“user”)：自动生成文件夹名称"runs"，并附带comment"user"，以及随机的events名称

二、writer.add_scalar()

使用add_scalar()将标量添加到event文件中：

writer = SummaryWriter("test0")
x = range(100)
for i in x:
    #writer.add_scalar(tag='y=2x', scalar_value=i * 2, global_step=i)
    writer.add_scalar('y=2x', i * 2, i)
writer.close()

参数：
$\bullet$tag(string)：数据标识符
$\bullet$scalar_value(float or string/blobname)：保存的数值(纵坐标y)
$\bullet$global_step(int)：时间步数(横坐标x)

三、tensorboard访问events

使用tensorboard --logdir=xxx访问 ( '='左右不要有空格 )：
浏览器访问 http://localhost:6006/：

四、参考

[1] https://pytorch.org/docs/stable/tensorboard.html