Handle tiny histograms, this is likely temporary...
@@ -215,6 +215,7 @@ def tf_summary_to_dict(tf_summary_str_or_pb, namespace=""):
# audio = wandb.Audio(six.BytesIO(value.audio.encoded_audio_string),
# sample_rate=value.audio.sample_rate, content_type=value.audio.content_type)
elif kind == "histo":
+ if len(value.histo.bucket_limit) >= 3:
first = value.histo.bucket_limit[0] + \
value.histo.bucket_limit[0] - value.histo.bucket_limit[1]
last = value.histo.bucket_limit[-2] + \
@@ -229,6 +230,8 @@ def tf_summary_to_dict(tf_summary_str_or_pb, namespace=""):
except ValueError:
wandb.termwarn("Not logging key \"{}\". Histograms must have fewer than {} bins".format(
tag, wandb.Histogram.MAX_LENGTH), repeat=False)
+ else:
+ wandb.termwarn("Not logging key \"{}\". Found an unsupported histogram format", repeat=False)
elif value.tag == "_hparams_/session_start_info":
if wandb.util.get_module("tensorboard.plugins.hparams"):
from tensorboard.plugins.hparams import plugin_data_pb2
