pytorch_lightning.trainer.callback_hook module¶
-
class
pytorch_lightning.trainer.callback_hook.
TrainerCallbackHookMixin
[source]¶ Bases:
abc.ABC
-
on_fit_start
(model)[source]¶ Called when the trainer initialization begins, model has not yet been set.
-
on_train_batch_start
(batch, batch_idx, dataloader_idx)[source]¶ Called when the training batch begins.
-
on_validation_batch_end
(batch, batch_idx, dataloader_idx)[source]¶ Called when the validation batch ends.
-