pytorch3->tensorboard的add_scalar函数基础应用
·
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对象,如果不关闭,程序中途崩溃,缓存里的数据可能丢失。
- 运行后,在终端执行:tensorboard --logdir=logs
更多推荐


所有评论(0)