LoggingCallback

class stable_pretraining.callbacks.LoggingCallback

Bases: Callback

Displays validation metrics in a color-coded formatted table.

This callback creates a visually appealing table of all validation metrics at the end of each validation epoch. Metrics are color-coded for better readability in terminal outputs.

Features: - Automatic sorting of metrics by name - Color coding: blue for metric names, green for values - Filters out internal metrics (log, progress_bar)

on_validation_end(trainer: Trainer, pl_module: LightningModule)

Called when the validation loop ends.