def write_pytorch_graph(self, graph_profile):
# https://github.com/pytorch/pytorch/blob/c749be9e9f8dd3db8b3582e93f917bd47e8e9e20/torch/utils/tensorboard/writer.py # L99
# graph_profile = pytorch_graph.graph(self.model)
graph = graph_profile[0]
stepstats = graph_profile[1]
event = Event(graph_def=graph.SerializeToString())
self.proto_writer.write_event(event)
trm = TaggedRunMetadata(tag="step1", run_metadata=stepstats.SerializeToString())
event = Event(tagged_run_metadata=trm)
self.proto_writer.write_event(event)
def write_summary_with_index(self,
summary,
step,
tname,
mode,
mode_step,
timestamp=None):
event = Event(summary=summary)
event.wall_time = timestamp if timestamp else time.time()
event.step = step
return self.write_event(EventWithIndex(event, tname, mode, mode_step))
def read_events(self, check="minimal"):
while self.has_data():
rec = self.read_record(check=check)
event = Event()
event.ParseFromString(rec)
yield event
def write_summary(self, summary, step, timestamp=None):
event = Event(summary=summary)
event.wall_time = timestamp if timestamp else time.time()
event.step = step
self.write_event(event)
def write_graph(self, graph):
"""Adds a `Graph` protocol buffer to the event file."""
event = Event(graph_def=graph.SerializeToString())
self.write_event(event)
def _get_sentinel_event():
"""Generate a sentinel event for terminating worker."""
return Event()
def write_summary(self, summary, step):
event = Event(summary=summary)
event.wall_time = time.time()
event.step = step
self.write_event(event)