pytorch_lightning.trainer.callback_hook module

class pytorch_lightning.trainer.callback_hook.TrainerCallbackHookMixin

Bases: abc.ABC

on_batch_end()

Called when the training batch ends.

on_batch_start()

Called when the training batch begins.

on_epoch_end()

Called when the epoch ends.

on_epoch_start()

Called when the epoch begins.

on_fit_end()

Called when the trainer initialization begins, model has not yet been set.

on_fit_start(model)

Called when the trainer initialization begins, model has not yet been set.

on_init_end()

Called when the trainer initialization ends, model has not yet been set.

on_init_start()

Called when the trainer initialization begins, model has not yet been set.

on_keyboard_interrupt()

Called when the training is interrupted by KeyboardInterrupt.

on_pretrain_routine_end(model)

Called when the train ends.

on_pretrain_routine_start(model)

Called when the train begins.

on_sanity_check_end()

Called when the validation sanity check ends.

on_sanity_check_start()

Called when the validation sanity check starts.

on_test_batch_end(batch, batch_idx, dataloader_idx)

Called when the test batch ends.

on_test_batch_start(batch, batch_idx, dataloader_idx)

Called when the test batch begins.

on_test_end()

Called when the test ends.

on_test_epoch_end()

Called when the epoch ends.

on_test_epoch_start()

Called when the epoch begins.

on_test_start()

Called when the test begins.

on_train_batch_end(batch, batch_idx, dataloader_idx)

Called when the training batch ends.

on_train_batch_start(batch, batch_idx, dataloader_idx)

Called when the training batch begins.

on_train_end()

Called when the train ends.

on_train_epoch_end()

Called when the epoch ends.

on_train_epoch_start()

Called when the epoch begins.

on_train_start()

Called when the train begins.

on_validation_batch_end(batch, batch_idx, dataloader_idx)

Called when the validation batch ends.

on_validation_batch_start(batch, batch_idx, dataloader_idx)

Called when the validation batch begins.

on_validation_end()

Called when the validation loop ends.

on_validation_epoch_end()

Called when the epoch ends.

on_validation_epoch_start()

Called when the epoch begins.

on_validation_start()

Called when the validation loop begins.

setup(stage)

Called in the beginning of fit and test

teardown(stage)

Called at the end of fit and test

callbacks: List[pytorch_lightning.callbacks.base.Callback] = []

get_model: Callable = None