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

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

用法:

add_scalars(main_tag, tag_scalar_dict, global_step=None, walltime=None)

参数：

• main_tag(string) -标签的父名称

• tag_scalar_dict(dict) -存储标签和对应值的键值对

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

• walltime(float) -事件纪元后的可选覆盖默认 walltime (time.time()) 秒

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

例子：

from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter()
r = 5
for i in range(100):
    writer.add_scalars('run_14h', {'xsinx':i*np.sin(i/r),
                                    'xcosx':i*np.cos(i/r),
                                    'tanx': np.tan(i/r)}, i)
writer.close()
# This call adds three values to the same scalar plot with the tag
# 'run_14h' in TensorBoard's scalar section.

预期结果：