pytorch使用tensorboard

from torch.utils.tensorboard import SummaryWriter
#log_dir参数：指定生成的日志文件保存的路径地址
writer = SummaryWriter(log_dir='runs/logs')
for i in range(100):

writer.add_scalar(tag="y=x*x", # 可以暂时理解为图像的名字
                  scalar_value=i * i,  # 纵坐标的值
                  global_step=i  # 横坐标
                  )

writer.close() # 关闭SummaryWriter

在控制台打开tensorboar

tensorboard --logdir=runs/logs

#展示runs/logs路径下的所有图片

在一幅图像中画多条函数曲线：

writer = SummaryWriter(log_dir='runs/log')
for i in range(100):
    writer.add_scalars('functions', {'y=x+20': i+20, 'y=x': i }, i)

此时会在run/log路径下生成两个文件夹：

打开tensorboard展示图如下：