Pytorch里用tensorboard将两条曲线画在一张图里面(报错:NotImplementedError: Got ＜class ‘dict‘＞, but numpy array, torch)

心中带点小风骚 • 2022年5月20日下午2:08 • 技术文章 • 阅读 900

上代码

from torch.utils.tensorboard import SummaryWriter

writer=SummaryWriter("log2")

for i in range(100):
     writer.add_scalars('haha', {'Train':i*i*3+0.5}, i)
     writer.add_scalars('haha', {'Test': i*i}, i)
writer.close()
#############################################
!!!是add_scalars而不是add_scalar
有一个s!
#############################################

否则报错:
请添加图片描述
NotImplementedError: Got <class ‘dict’>, but numpy array, torch tensor, or caffe2 blob name are expected.