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

最新推荐文章于 2024-09-11 19:05:57 发布

嘉陵妹妹

最新推荐文章于 2024-09-11 19:05:57 发布

阅读量4.8k

点赞数 15

CC 4.0 BY-SA版权

文章标签： pytorch 深度学习 python

本文链接：https://blog.youkuaiyun.com/X131644/article/details/124860810

本文介绍了如何使用PyTorch的TensorBoard工具，通过`add_scalars`方法记录训练和测试过程中的多个指标。示例代码展示了如何每100步添加一次'哈哈'标签下的'Train'和'Test'值。务必注意`add_scalars`函数的正确用法，以避免因传入非预期类型参数导致的错误。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

上代码

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