flambe.logging.handler.tensorboard

Module Contents

class flambe.logging.handler.tensorboard.TensorboardXHandler(*args: Any, **kwargs: Any)[source]

Bases: logging.Handler

Implements Tensorboard message logging via TensorboardX

Parameters:
  • writer (SummaryWriter) – Initialized TensorboardX Writer
  • *args (Any) – Other positional args for logging.Handler
  • **kwargs (Any) – Other kwargs for logging.Handler
writer

Initialized TensorboardX Writer

Type:SummaryWriter
emit(self, record: logging.LogRecord)[source]

Save to tensorboard logging directory

Overrides logging.Handler.emit

Parameters:record (logging.LogRecord) – LogRecord with data relevant to Tensorboard
Returns:
Return type:None
close(self)[source]

Teardown writers and teardown super

Returns:
Return type:None
flush(self)[source]

Call flush on the Tensorboard writer

Returns:
Return type:None