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), )