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()

預期結果：