from torch.utils.tensorboard import SummaryWriter

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

1.from torch.utils.tensorboard import SummaryWriter

torch:PyTorch 核心库。

utils:PyTorch 实用工具包。

tensorboard:模块,张量板,让训练过程显示在浏览器上。

SummaryWriter:类,负责把数据写进日志文件,供tensorboard使用

2.writer = SummaryWriter(“logs”)

实例化SummaryWriter类的对象writer,将TensorBoard 的动作存入文件’logs‘

3.for i in range(100)

从0到99取100个数

4.writer.add_scalar(“y=2x”, 2 * i, i)

scalar:标量,有别于向量,它是无方向的量。

add_scalar:挨个描点,启动TensorBoard 后渲染成线

writer.add_scalar(“y=2x”, 2 * i, i)
        # ↑ ↑ ↑
        # tag value step
        # 标题 Y轴的值 X轴的值
5.writer.close()

关闭writer对象,如果不关闭,程序中途崩溃,缓存里的数据可能丢失。

  1. 运行后,在终端执行:tensorboard --logdir=logs
Logo

Agent 垂直技术社区,欢迎活跃、内容共建。

更多推荐