是否可以根据其值省略Tensorflow标量摘要?

时间:2018-02-23 10:08:25

标签: tensorflow tensorboard

我构建摘要操作并将它们添加到集合中,然后在培训/验证期间始终将摘要集合评估为sess.run调用的一部分。

但是,在某些情况下,值为nan,这会使Tensorboard图表变坏。 (三角形而不是数据点,并且平滑不会在两者之间使用纳米值。)

是否有办法从集合中省略特定摘要,具体取决于有效值?我可以用零或类似的方法替换nan值,但任何人为选择的值都会污染真实报告的统计数据。

我添加如下摘要: tf.summary.scalar('scc_precision_test', precision_test, [Constants.TEST_SUMMARIES])

谢谢!

1 个答案:

答案 0 :(得分:5)

您可以在将摘要写入FileWriter

之前检查摘要的价值
prec_test = tf.summary.scalar('scc_precision_test', precision_test,
                              [Constants.TEST_SUMMARIES])
# ... 
..., prec_test_sum = sess.run([..., prec_test], ...)
prec_test_sum = tf.Summary().FromString(prec_test_sum)
if np.isfinite(prec_test_sum.value[0].simple_value):
    writer.add_summary(prec_test_sum.SerializeToString(), global_step=...)

如果您将多个摘要合并到一个tf.Summary对象中(例如,使用tf.summary.merge / tf.summary.merge_all制作),那么您必须过滤value字段:< / p>

prec_test = tf.summary.scalar('scc_precision_test', precision_test,
                              [Constants.TEST_SUMMARIES])
merged = tf.summary.merge_all(key=Constants.TEST_SUMMARIES)
# ... 
..., merged_sum = sess.run([..., merged], ...)
merged_sum = tf.Summary().FromString(merged_sum)
# Reversed traversal to be able to delete elements while iterating correctly
for i, value in reversed(list(enumerate(merged_sum.value))):
    # Discard summary if is scalar and not finite
    if value.WhichOneof('value') == 'simple_value' and not np.isfinite(value.simple_value):
        del merged_sum[i]
# Write all valid summaries
writer.add_summary(merged_sum.SerializeToString(), global_step=...)