kiwi.training.callbacks

Module Contents

Classes

BestMetricsInfo

Class for logging current training metrics along with the best so far.

kiwi.training.callbacks.logger

class kiwi.training.callbacks.BestMetricsInfo(monitor: str = 'val_loss', min_delta: float = 0.0, verbose: bool = True, mode: str = 'auto')

Bases: pytorch_lightning.Callback

Class for logging current training metrics along with the best so far.

on_train_begin(self, trainer, pl_module)

on_train_end(self, trainer, pl_module)

Called when the train ends.

on_validation_end(self, trainer, pl_module)

Called when the validation loop ends.

kiwi.training kiwi.training.optimizers