您可以在将摘要写入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 字段:
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=...)