Python PyTorch SummaryWriter.add_text用法及代碼示例

本文簡要介紹python語言中 torch.utils.tensorboard.writer.SummaryWriter.add_text 的用法。

用法:

add_text(tag, text_string, global_step=None, walltime=None)

參數：

• tag(string) -數據標識符

• text_string(string) -要保存的字符串

• global_step(int) -要記錄的全局步長 值

• walltime(float) -事件紀元後的可選覆蓋默認 walltime (time.time()) 秒

將文本數據添加到摘要中。

例子：

writer.add_text('lstm', 'This is an lstm', 0)
writer.add_text('rnn', 'This is an rnn', 10)