我构建摘要操作并将它们添加到集合中,然后在培训/验证期间始终将摘要集合评估为sess.run
调用的一部分。
但是,在某些情况下,值为nan
,这会使Tensorboard图表变坏。 (三角形而不是数据点,并且平滑不会在两者之间使用纳米值。)
是否有办法从集合中省略特定摘要,具体取决于有效值?我可以用零或类似的方法替换nan值,但任何人为选择的值都会污染真实报告的统计数据。
我添加如下摘要:
tf.summary.scalar('scc_precision_test', precision_test, [Constants.TEST_SUMMARIES])
谢谢!
答案 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=...)