Skip to content

Progress Bar Callback

Source

Progressbar callback.

ProgressBarCallback

Bases: TQDMProgressBar

Progress bar for training and validation steps.

get_metrics(trainer, pl_module)

Override this to customize the metrics displayed in the progress bar.

Parameters:

  • trainer (Trainer) –

    The trainer object.

  • pl_module (LightningModule) –

    The LightningModule object, unused.

Returns:

  • dict –

    A dictionary with the metrics to display in the progress bar.

init_test_tqdm()

Override this to customize the tqdm bar for testing.

Returns:

  • tqdm –

    A tqdm bar.

init_train_tqdm()

Override this to customize the tqdm bar for training.

Returns:

  • tqdm –

    A tqdm bar.

init_validation_tqdm()

Override this to customize the tqdm bar for validation.

Returns:

  • tqdm –

    A tqdm bar.