pytorch Tensorboard的使用

标量数据可视化

import numpy as np
from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter(log_dir='./runs')  #log_dir用于指定保存目录
flag = 1


if flag:
    for x in range(100):
        # 把x*2的数据加入标签y=2x的曲线
        writer.add_scalar(tag='y=2x', scalar_value=x * 2, global_step=x)

        # 把2**x的数据加入标签y=pow(2,x)的曲线
        writer.add_scalar(tag='y=pow(2,x)', scalar_value=2 ** x,global_step=x)

        # 把x*sin(x)和x*cos(x)的数据加入data/scalar_group的标签组中，即
        # 两个曲线绘制在一张图中
        writer.add_scalars('data/scalar_group',{'xsinx': x * np.sin(x), 'xcosx': x * np.cos(x)}, x)
writer.close()  # 将event log写完之后，记得close()