experiment.tensorboard_x module

class experiment.tensorboard_x.TensorBoardXLogHandler(summary_writer, title='Logging', *args, **kwds)

Bases: logging.Handler

Logging handler that logs to a TensorBoardX instance.

Parameters:

• summary_writer (tensorboard.SummaryWriter) – The summarywriter to log to.
• title (string) – Title/tag to write to.

emit(record)

Do whatever it takes to actually log the specified logging record.

This version is intended to be implemented by subclasses and so raises a NotImplementedError.

experiment.tensorboard_x.monitor_gpu(summary_writer, gpu_index=None, xtick_size=100)

Monitor the memory and utilization of a GPU.

Parameters:

• env (str) – The visdom environment to log to.
• gpu_index (int) – The GPU to monitor.

experiment.tensorboard_x.write_conf(summary_writer, args=None, text=None)

Write configuration to the Visdom env.

Parameters:

• summary_writer (tensorboard.SummaryWriter) – The summarywriter to log to.
• args (Namespace, optional) – The argument namespace returned by argparse.
• text (string, optional) – Configuration as text block.