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.
"""
|