Skip to content
SR-Forge

Checkpointing

PyTorchModelSaver

Bases: Observer

Observer that saves model checkpoints to disk after each epoch.

Tracks the best validation loss and saves both _best and _last checkpoints (training state + model weights per sub-model).

Parameters:

Name Type Description Default
tracker ExperimentTracker | None

Experiment tracker for uploading files and setting summaries. Defaults to :class:~srforge.tracking.NullTracker if not provided.

None
best_loss float

Initial best loss value (used when resuming).

None