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)