from pytorch_lightning.callbacks.callback import Callback class Checkpoint(Callback): r"""This is the base class for model checkpointing. Expert users may want to subclass it in case of writing custom :class:`~pytorch_lightning.callbacks.Checkpoint` callback, so that the trainer recognizes the custom class as a checkpointing callback. """