Callback¶
-
class
pytorch_lightning.callbacks.
Callback
[source]¶ Bases:
abc.ABC
Abstract base class used to build new callbacks.
Subclass this class and override any of the relevant hooks
-
on_init_end
(trainer)[source]¶ Called when the trainer initialization ends, model has not yet been set.
-
on_init_start
(trainer)[source]¶ Called when the trainer initialization begins, model has not yet been set.
-
on_keyboard_interrupt
(trainer, pl_module)[source]¶ Called when the training is interrupted by KeyboardInterrupt.
-
on_load_checkpoint
(checkpointed_state)[source]¶ Called when loading a model checkpoint, use to reload state.
-
on_save_checkpoint
(trainer, pl_module)[source]¶ Called when saving a model checkpoint, use to persist state.
-
on_test_batch_end
(trainer, pl_module, outputs, batch, batch_idx, dataloader_idx)[source]¶ Called when the test batch ends.
-
on_test_batch_start
(trainer, pl_module, batch, batch_idx, dataloader_idx)[source]¶ Called when the test batch begins.
-
on_train_batch_end
(trainer, pl_module, outputs, batch, batch_idx, dataloader_idx)[source]¶ Called when the train batch ends.
-
on_train_batch_start
(trainer, pl_module, batch, batch_idx, dataloader_idx)[source]¶ Called when the train batch begins.
-
on_validation_batch_end
(trainer, pl_module, outputs, batch, batch_idx, dataloader_idx)[source]¶ Called when the validation batch ends.
-