LambdaCallback

class braintools.trainer.LambdaCallback(**kwargs)

Create a callback from lambda functions.

Parameters:

**kwargs – Hook names mapped to callable functions.

Examples

>>> callback = LambdaCallback(
...     on_train_epoch_end=lambda trainer, module: print(f"Epoch {module.current_epoch} done!")
... )
>>> trainer = Trainer(callbacks=[callback])