[教训贴]Pytorch中使用tensorboard报错NotImplementedError: Got ＜class ‘dict‘＞, but numpy array, torch tensor
报错信息：报错原因：writer.add_scalar要加“S”
·
报错信息：
NotImplementedError: Got <class ‘dict’>, but numpy array, torch tensor, or caffe2 blob name are expected.
报错原因：
writer.add_scalars要加“S”
单个变量的可视化是不用加s的，例如：
- writer.add_scalar(‘test_acc’, test_acc, test_step)
多个变量共同可视化就要加S，例如： - writer.add_scalars(
‘epoch_loss’,
{
‘train’: np.mean(train_loss_lis),
‘test’: np.mean(test_loss_lis)
},
epoch
)
最终生成的效果图：
分析：就是自己太不仔细了，也太依赖编译器，出来啥就点啥了
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
5
0- 0
已为社区贡献5条内容
所有评论(0)