quaterion.train.callbacks.metrics_callback module¶

class MetricsCallback(*_: Any, **__: Any)[source]¶

Bases: Callback

on_sanity_check_end(trainer: Trainer, trainable_model: LightningModule) → None[source]¶

on_train_epoch_end(trainer: Trainer, trainable_model: LightningModule) → None[source]¶

on_train_epoch_start(trainer: Trainer, trainable_model: LightningModule) → None[source]¶

on_validation_epoch_end(trainer: Trainer, trainable_model: LightningModule) → None[source]¶

on_validation_epoch_start(trainer: Trainer, trainable_model: LightningModule) → None[source]¶

static reset_metrics(trainable_model)[source]¶