LambdaCallback — PyTorch Lightning 2.5.1.post0 documentation (original) (raw)

class lightning.pytorch.callbacks.LambdaCallback(setup=None, teardown=None, on_fit_start=None, on_fit_end=None, on_sanity_check_start=None, on_sanity_check_end=None, on_train_batch_start=None, on_train_batch_end=None, on_train_epoch_start=None, on_train_epoch_end=None, on_validation_epoch_start=None, on_validation_epoch_end=None, on_test_epoch_start=None, on_test_epoch_end=None, on_validation_batch_start=None, on_validation_batch_end=None, on_test_batch_start=None, on_test_batch_end=None, on_train_start=None, on_train_end=None, on_validation_start=None, on_validation_end=None, on_test_start=None, on_test_end=None, on_exception=None, on_save_checkpoint=None, on_load_checkpoint=None, on_before_backward=None, on_after_backward=None, on_before_optimizer_step=None, on_before_zero_grad=None, on_predict_start=None, on_predict_end=None, on_predict_batch_start=None, on_predict_batch_end=None, on_predict_epoch_start=None, on_predict_epoch_end=None)[source]¶

Bases: Callback

Create a simple callback on the fly using lambda functions.

Parameters:

**kwargs¶ – hooks supported by Callback

Example:

from lightning.pytorch import Trainer from lightning.pytorch.callbacks import LambdaCallback trainer = Trainer(callbacks=[LambdaCallback(setup=lambda *args: print('setup'))])