编程语言
首页 > 编程语言> > 【编程积累】在tensorboard里绘制多条曲线

【编程积累】在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类可以打开新的文件,也可以用来修改已经存在的文件

标签:SummaryWriter,绘制,编程,value,step,tensorboard,line,tb
来源: https://www.cnblogs.com/mhlan/p/15932556.html