Python PyTorch SummaryWriter.add_scalar用法及代码示例

本文简要介绍python语言中 torch.utils.tensorboard.writer.SummaryWriter.add_scalar 的用法。

用法:

add_scalar(tag, scalar_value, global_step=None, walltime=None, new_style=False, double_precision=False)

参数：

• tag(string) -数据标识符

• scalar_value(float或者字符串/blob 名称) -保存值

• global_step(int) -要记录的全局步长 值

• walltime(float) -可选的覆盖默认的墙时间(时间。time())在事件纪元之后的秒数

• new_style(布尔值) -是使用新样式(张量字段)还是旧样式(simple_value 字段)。新样式可能会导致更快的数据加载。

将标量数据添加到摘要中。

例子：

from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter()
x = range(100)
for i in x:
    writer.add_scalar('y=2x', i * 2, i)
writer.close()

预期结果：