def _WriteToLog(self, text, logdir, filename):
"""Logs `text` and saves it under `logdir/filename`."""
with tf.gfile.GFile(os.path.join(logdir, filename), 'w') as f:
f.write(text)
if self._summary_writer is not None:
# Force newlines to be rendered correctly by Markdown.
text = text.replace('\n', ' \n')
self._summary_writer.add_summary(
tf.Summary(value=[
tf.Summary.Value(tag=filename,
tensor=tf.make_tensor_proto([text]))
]))
def SerializeOutputs(nmap: py_utils.NestedMap) -> bytes:
"""Return a serialized representation of the contents of `nmap`.
Args:
nmap: A NestedMap of data to serialize.
Returns:
A serialized record_pb2.Record() of the contents of `nmap`.
"""
record = record_pb2.Record()
flat_nmap = nmap.FlattenItems()
for key, value in flat_nmap:
record.fields[key].CopyFrom(tf.make_tensor_proto(value))
serialized = record.SerializeToString()
return serialized
def AddKeyVals(**kwargs):
record = record_pb2.Record()
for k, v in kwargs.items():
record.fields[k].CopyFrom(tf.make_tensor_proto(v))
return record
