File size: 361 Bytes
9c6594c
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
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.

    """