trw.callbacks.callback_tensorboard

Module Contents

Classes

CallbackTensorboardBased	Tensorboard based callback. Manages a single tensorboardX.SummaryWriter instance
CallbackClearTensorboardLog	Remove any existing logger

Attributes

logger
with_tensorboardX
with_tensorboardX

trw.callbacks.callback_tensorboard.logger

trw.callbacks.callback_tensorboard.with_tensorboardX = True

trw.callbacks.callback_tensorboard.with_tensorboardX = False

class trw.callbacks.callback_tensorboard.CallbackTensorboardBased

Bases: trw.callbacks.callback.Callback

Tensorboard based callback. Manages a single tensorboardX.SummaryWriter instance

_tensorboard_logger

static create_logger(path)

Create a tensorboardX.SummaryWriter instance. If an instance already exists or tensorboardX could not be imported, no logger will be created :param path: where to write the tensorboard log :return: a logger or None if logger creation failed

static get_tensorboard_logger()

Returns

None if the tensorboad logger was not created or a tensorboardX.SummaryWriter

static remove_tensorboard_logger()

Remove the current tensorboardX.SummaryWriter

class trw.callbacks.callback_tensorboard.CallbackClearTensorboardLog

Bases: CallbackTensorboardBased

Remove any existing logger

This is useful when we train multiple models so that they have their own tensorboard log file

__call__(self, options, history, model, losses, outputs, datasets, datasets_infos, callbacks_per_batch, **kwargs)