espnet.utils.training.tensorboard_logger.TensorboardLogger

source

class espnet.utils.training.tensorboard_logger.TensorboardLogger(logger, att_reporter=None, ctc_reporter=None, entries=None, epoch=0)

Bases: Extension

A tensorboard logger extension.

Init the extension.

• Parameters:
  • logger (SummaryWriter) – The logger to use
  • att_reporter (PlotAttentionReporter) – The (optional) PlotAttentionReporter
  • entries – The entries to watch
  • epoch (int) – The starting epoch

default_name = 'espnet_tensorboard_logger'