quaterion.train.callbacks.metrics_callback module

class MetricsCallback(*args: Any, **kwargs: Any)

Bases: Callback

on_sanity_check_end(trainer: Trainer, trainable_model: LightningModule) → None

Called when the validation sanity check ends.

on_train_epoch_end(trainer: Trainer, trainable_model: LightningModule) → None

Called when the train epoch ends.

To access all batch outputs at the end of the epoch, either:

1. Implement training_epoch_end in the LightningModule and access outputs via the module OR

2. Cache data across train batch hooks inside the callback implementation to post-process in this hook.

on_train_epoch_start(trainer: Trainer, trainable_model: LightningModule) → None

Called when the train epoch begins.

on_validation_epoch_end(trainer: Trainer, trainable_model: LightningModule) → None

Called when the val epoch ends.

on_validation_epoch_start(trainer: Trainer, trainable_model: LightningModule) → None

Called when the val epoch begins.

static reset_metrics(trainable_model)