def __init__(self, flow: TaskFlow, tensorboard_converters: Optional[TensorboardConverters] = None):
super().__init__(CallbackOrder.Metric)
self.flow = flow
self.overall_loss = flow.get_per_sample_loss()
self.leaf_losses = self.overall_loss.get_leaf_losses_per_sample()
self.interpretations = {}
self.loader_counts = {}
self.tensorboard_converters = tensorboard_converters
def __init__(
self,
flow: TaskFlow,
tensorboard_converters: Optional[TensorboardConverters] = None,
loaders_to_skip=()):
"""
:param flow: The task flow, which holds the per sample loss functions for every task.
:param tensorboard_converters: A :class:`TensorboardConverters` object which is responsible for the Tensorboard
logging settings.
:param loaders_to_skip: Optional loaders to be skipped, for example because labels aren't available for them.
"""
super().__init__(CallbackOrder.Metric)
self.flow = flow
self.loaders_to_skip = loaders_to_skip
self.overall_loss = flow.get_per_sample_loss()
self.leaf_losses = self.overall_loss.get_leaf_losses_per_sample()
self.interpretations = {}
self.loader_counts = {}
self.tensorboard_converters = tensorboard_converters