Tensorboards add_scalar 仅记录在 for 循环中分配的最后一个值

Question

我正在尝试使用 add_scalar 方法将存储在 dict 对象中的 ML 模型的阈值相关精度值记录到 Tensorboard。但是，它只会将最后一个数据点添加到图形中。

字典基本上看起来是这样的：

eval_metrics['accuracy'] = {0.2: 0.375,
 0.4: 0.8,
 0.6: 0.75,
 0.8: 0.35}

我的 for 循环看起来像这样：

for key in eval_metrics['accuracy']: 
    writer.add_scalar('acc', eval_metrics['accuracy'][key], key)

但是，只有四个键值对中的最后一个被注册，因此生成的标量只包含一个数据点。有谁知道这是为什么，或者是否有解决办法？

我已经为 global_step 值尝试了不同的数据类型，或者引入了一个新变量来替换 key 变量。

Answer 1

add_scalars 而不是 add_scalar 是更适合您需要的功能。