Progress Bar Callback
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:
| Name | Type | Description | Default |
|---|---|---|---|
trainer
|
Trainer
|
The trainer object. |
required |
pl_module
|
LightningModule
|
The LightningModule object, unused. |
required |
Returns:
| Type | Description |
|---|---|
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:
| Type | Description |
|---|---|
tqdm
|
A tqdm bar. |
init_train_tqdm()
Override this to customize the tqdm bar for training.
Returns:
| Type | Description |
|---|---|
tqdm
|
A tqdm bar. |
init_validation_tqdm()
Override this to customize the tqdm bar for validation.
Returns:
| Type | Description |
|---|---|
tqdm
|
A tqdm bar. |