OnExceptionCheckpoint

class lightning.pytorch.callbacks.OnExceptionCheckpoint(dirpath, filename='on_exception')[source]

Bases: Checkpoint

Used to save a checkpoint on exception.

Parameters:
  • dirpath (Union[str, Path]) – directory to save the checkpoint file.

  • filename (str) – checkpoint filename. This must not include the extension.

Raises:

ValueError – If filename is empty.

Example

>>> from lightning.pytorch import Trainer
>>> from lightning.pytorch.callbacks import OnExceptionCheckpoint
>>> trainer = Trainer(callbacks=[OnExceptionCheckpoint(".")])
on_exception(trainer, *_, **__)[source]

Called when any trainer execution is interrupted by an exception.

Return type:

None

teardown(trainer, *_, **__)[source]

Called when fit, validate, test, predict, or tune ends.

Return type:

None