RichProgressBar

class braintools.trainer.RichProgressBar(refresh_rate=1, leave=False)

Rich progress bar for training visualization.

Requires the ‘rich’ package to be installed.

Parameters:

• refresh_rate (int) – Number of batches between progress bar updates.

• leave (bool) – Whether to leave the progress bar after completion.

Examples

>>> progress = RichProgressBar()
>>> trainer = Trainer(callbacks=[progress], enable_progress_bar=False)

on_train_batch_end(trainer, module, outputs, batch, batch_idx)

Update progress bar.

on_train_epoch_end(trainer, module)

Complete progress bar.

on_train_epoch_start(trainer, module)

Create progress bar for epoch.