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

上代码

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!
#############################################

否则报错:
Pytorch里用tensorboard将两条曲线画在一张图里面(报错:NotImplementedError: Got <class ‘dict‘>, but numpy array, torch)_第1张图片
NotImplementedError: Got , but numpy array, torch tensor, or caffe2 blob name are expected.

运行截图
Pytorch里用tensorboard将两条曲线画在一张图里面(报错:NotImplementedError: Got <class ‘dict‘>, but numpy array, torch)_第2张图片

你可能感兴趣的:(pytorch,深度学习,python)