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