quaterion.utils.progress_bar module

class FixedLengthProcessionSpeed(style: str | Any)

Bases: ProcessingSpeedColumn

Renders processing speed for the progress bar with fixes length

render(task) → ConsoleRenderable | RichCast | str

Should return a renderable object.

class QuaterionProgressBar(refresh_rate: int = 1, leave: bool = False, theme: RichProgressBarTheme | None = None, console_kwargs: Dict[str, Any] | None = None)

Bases: RichProgressBar

configure_columns(trainer) → list

on_predict_batch_start(trainer: Trainer, pl_module: LightningModule, batch: Any, batch_idx: int, dataloader_idx: int) → None

Called when the predict batch begins.

on_predict_end(trainer: Trainer, pl_module: LightningModule) → None

Called when predict ends.

property predict_description: str