【编程积累】在tensorboard里绘制多条曲线

代码

from torch.utils.tensorboard import SummaryWriter

for line in range(10):
    tb = SummaryWriter('./runs/lines/' + str(line))
    for step in range(10):
        value = step + line
        tb.add_scalar('value of lines', value, step)
    tb.close()

结果

image-20220224165040480

注意事项

SummaryWriter类可以打开新的文件,也可以用来修改已经存在的文件

posted @ 2022-02-24 16:53  木坑  阅读(778)  评论(0编辑  收藏  举报