callcut.training.LoggingCallback

class callcut.training.LoggingCallback

Callback that logs training progress using the callcut logger.

This callback provides formatted logging at the end of each epoch, complementing Lightning’s built-in progress bar.

Methods

on_train_epoch_end(trainer, pl_module)

Log training metrics at the end of each epoch.

Examples

>>> from callcut.training import LoggingCallback
>>> import lightning as L
>>>
>>> trainer = L.Trainer(
...     max_epochs=10,
...     callbacks=[LoggingCallback()],
... )

on_train_epoch_end(trainer, pl_module)

Log training metrics at the end of each epoch.

Parameters:

trainerTrainer

The Lightning trainer instance.

pl_moduleLightningModule

The Lightning module being trained.