def setUp(self): # Populate the log directory with debugger event for run '.'. self.log_dir = self.get_temp_dir() file_prefix = compat.as_bytes( os.path.join(self.log_dir, 'events.debugger')) writer = pywrap_tensorflow.EventsWriter(file_prefix) writer.WriteEvent( self._CreateEventWithDebugNumericSummary(op_name='layers/Matmul', output_slot=0, wall_time=42, step=2, list_of_values=[1, 2, 3])) writer.WriteEvent( self._CreateEventWithDebugNumericSummary(op_name='layers/Matmul', output_slot=1, wall_time=43, step=7, list_of_values=[4, 5, 6])) writer.WriteEvent( self._CreateEventWithDebugNumericSummary(op_name='logits/Add', output_slot=0, wall_time=1337, step=7, list_of_values=[7, 8, 9])) writer.WriteEvent( self._CreateEventWithDebugNumericSummary( op_name='logits/Add', output_slot=0, wall_time=1338, step=8, list_of_values=[10, 11, 12])) writer.Close() # Populate the log directory with debugger event for run 'run_foo'. run_foo_directory = os.path.join(self.log_dir, 'run_foo') os.mkdir(run_foo_directory) file_prefix = compat.as_bytes( os.path.join(run_foo_directory, 'events.debugger')) writer = pywrap_tensorflow.EventsWriter(file_prefix) writer.WriteEvent( self._CreateEventWithDebugNumericSummary( op_name='layers/Variable', output_slot=0, wall_time=4242, step=42, list_of_values=[13, 14, 15])) writer.Close() # Start a server that will receive requests and respond with health pills. self.multiplexer = event_multiplexer.EventMultiplexer({ '.': self.log_dir, 'run_foo': run_foo_directory, }) self.plugin = debugger_plugin.DebuggerPlugin() wsgi_app = application.TensorBoardWSGIApp(self.log_dir, [self.plugin], self.multiplexer, reload_interval=0) self.server = werkzeug_test.Client(wsgi_app, wrappers.BaseResponse)
def standard_tensorboard_wsgi(logdir, purge_orphaned_data, reload_interval): """Construct a TensorBoardWSGIApp with standard plugins and multiplexer.""" multiplexer = event_multiplexer.EventMultiplexer( size_guidance=DEFAULT_SIZE_GUIDANCE, purge_orphaned_data=purge_orphaned_data) plugins = { debugger_plugin.PLUGIN_PREFIX_ROUTE: debugger_plugin.DebuggerPlugin(), projector_plugin.PLUGIN_PREFIX_ROUTE: projector_plugin.ProjectorPlugin(), } return TensorBoardWSGIApp(logdir, plugins, multiplexer, reload_interval)
def setUp(self): self.fake_event_multiplexer = FakeEventMultiplexer({ '.': { 'layers/Matmul': [ event_accumulator.HealthPillEvent( wall_time=42, step=2, node_name='layers/Matmul', output_slot=0, value=[1, 2, 3]), event_accumulator.HealthPillEvent( wall_time=43, step=3, node_name='layers/Matmul', output_slot=1, value=[4, 5, 6]), ], 'logits/Add': [ event_accumulator.HealthPillEvent(wall_time=1337, step=7, node_name='logits/Add', output_slot=0, value=[7, 8, 9]), event_accumulator.HealthPillEvent(wall_time=1338, step=8, node_name='logits/Add', output_slot=0, value=[10, 11, 12]), ], }, 'run_foo': { 'layers/Variable': [ event_accumulator.HealthPillEvent( wall_time=4242, step=42, node_name='layers/Variable', output_slot=0, value=[13, 14, 15]), ], }, }) self.debugger_plugin = debugger_plugin.DebuggerPlugin( self.fake_event_multiplexer) self.unused_run_paths = {} self.unused_logdir = '/logdir'