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'
