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.

預期結果：