def _HParamSessionEnd(status):
from tensorboard.plugins.hparams import summary_v2 as hp
info = hp.plugin_data_pb2.SessionEndInfo(status=_Status(status))
return hp._summary_pb(
hp.metadata.SESSION_END_INFO_TAG,
hp.plugin_data_pb2.HParamsPluginData(session_end_info=info),
)
def _legacy_hparams_pb(hparams, trial_id):
from tensorboard.plugins.hparams import summary_v2 as hp
hparams = hp._normalize_hparams(hparams)
info = hp.plugin_data_pb2.SessionStartInfo(group_name=trial_id)
for name in sorted(hparams):
val = hparams[name]
if isinstance(val, bool):
info.hparams[name].bool_value = val
elif isinstance(val, (float, int)):
info.hparams[name].number_value = val
elif isinstance(val, six.string_types):
info.hparams[name].string_value = val
elif val is None:
info.hparams[name].string_value = ""
else:
info.hparams[name].string_value = str(val)
return hp._summary_pb(
hp.metadata.SESSION_START_INFO_TAG,
hp.plugin_data_pb2.HParamsPluginData(session_start_info=info),
)
def LegacySession(group_name, hparams):
import time
hparams = hp._normalize_hparams(hparams)
ssi = hp.plugin_data_pb2.SessionStartInfo(
group_name=group_name,
start_time_secs=time.time(),
)
for name in sorted(hparams):
val = hparams[name]
if isinstance(val, bool):
ssi.hparams[name].bool_value = val
elif isinstance(val, (float, int)):
ssi.hparams[name].number_value = val
elif isinstance(val, six.string_types):
ssi.hparams[name].string_value = val
else:
assert False, (name, val)
return hp._summary_pb(
hp.metadata.SESSION_START_INFO_TAG,
hp.plugin_data_pb2.HParamsPluginData(session_start_info=ssi),
)