Toggle Light / Dark / Auto color theme

Toggle table of contents sidebar

finetuner.tuner.callback.progress_bar module

class finetuner.tuner.callback.progress_bar.ProgressBarCallback

Bases: finetuner.tuner.callback.base.BaseCallback

A progress bar callback, using the rich progress bar.

property train_loss_str: str

Return type

str

property val_loss_str: str

Return type

str

on_fit_begin(tuner)

Called at the start of the fit method call, after all setup has been done, but before the training has started.

on_train_epoch_begin(tuner)

Called at the begining of training part of the epoch.

on_train_batch_end(tuner)

Called at the end of a training batch, after the backward pass.

on_val_begin(tuner)

Called at the start of the evaluation.

on_val_batch_end(tuner)

Called at the end of an evaluation batch.

on_val_end(tuner)

Called at the end of the evaluation.

on_fit_end(tuner)

Called at the end of the fit method call, after finishing all the epochs.

on_exception(tuner, exception)

Called when the tuner encounters an exception during execution.

on_keyboard_interrupt(tuner)

Called when the tuner is interrupted by the user