pytorch_lightning.callbacks.base module

Callback Base

Abstract base class used to build new callbacks.

class pytorch_lightning.callbacks.base.Callback

Bases: abc.ABC

Abstract base class used to build new callbacks.

on_batch_end(trainer, pl_module)

Called when the training batch ends.

on_batch_start(trainer, pl_module)

Called when the training batch begins.

on_epoch_end(trainer, pl_module)

Called when the epoch ends.

on_epoch_start(trainer, pl_module)

Called when the epoch begins.

on_fit_end(trainer)

Called when fit ends

on_fit_start(trainer)

Called when fit begins

on_init_end(trainer)

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

on_init_start(trainer)

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

on_keyboard_interrupt(trainer, pl_module)

Called when the training is interrupted by KeyboardInterrupt.

on_sanity_check_end(trainer, pl_module)

Called when the validation sanity check ends.

on_sanity_check_start(trainer, pl_module)

Called when the validation sanity check starts.

on_test_batch_end(trainer, pl_module)

Called when the test batch ends.

on_test_batch_start(trainer, pl_module)

Called when the test batch begins.

on_test_end(trainer, pl_module)

Called when the test ends.

on_test_start(trainer, pl_module)

Called when the test begins.

on_train_end(trainer, pl_module)

Called when the train ends.

on_train_start(trainer, pl_module)

Called when the train begins.

on_validation_batch_end(trainer, pl_module)

Called when the validation batch ends.

on_validation_batch_start(trainer, pl_module)

Called when the validation batch begins.

on_validation_end(trainer, pl_module)

Called when the validation loop ends.

on_validation_start(trainer, pl_module)

Called when the validation loop begins.

setup(trainer, stage)

Called when fit or test begins

teardown(trainer, stage)

Called when fit or test ends