def add_summary(self, summary, global_step=None):
'''Add a `Summary` protocol buffer to the event file.'''
if isinstance(summary, bytes):
summ = summary_pb2.Summary()
summ.ParseFromString(summary)
summary = summ
event = event_pb2.Event(summary=summary)
self._add_event(event, global_step)
def __init__(self, file_prefix='events'):
'''
Events files have a name of the form
'/some/file/path/events.out.tfevents.[timestamp].[hostname]'
'''
self._path = file_prefix + ".out.tfevents." + \
str(time.time())[:10] + \
"." + socket.gethostname()
self._num_outstanding_events = 0
# X: open up the writer
self._writer = open(self._path, 'wb')
# Initialize an event instance.
self._event = event_pb2.Event()
# X: get the time
self._event.wall_time = time.time()
# write event
self.write_event(self._event)
def add_graph(self, graph):
'''Add a `Graph` protocol buffer to the event file.'''
event = event_pb2.Event(graph_def=graph.SerializeToString())
self._add_event(event, None)