参考链接: add_text
代码:
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter()
for i in range(3):
writer.add_text('lstm', 'This is an lstm', i)
writer.add_text('rnn', 'This is an rnn', 10+i)
writer.close()
# tensorboard --logdir=runs
实验结果展示: