finetuner.tuner.callback.progress_bar module#

class finetuner.tuner.callback.progress_bar.ProgressBarCallback[source]#

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)[source]#

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)[source]#

Called at the begining of training part of the epoch.

on_train_batch_end(tuner)[source]#

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

on_val_begin(tuner)[source]#

Called at the start of the evaluation.

on_val_batch_end(tuner)[source]#

Called at the end of an evaluation batch.

on_val_end(tuner)[source]#

Called at the end of the evaluation.

on_fit_end(tuner)[source]#

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

on_exception(tuner, exception)[source]#

Called when the tuner encounters an exception during execution.

on_keyboard_interrupt(tuner)[source]#

Called when the tuner is interrupted by the user