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: |
None
|
best_loss
|
float
|
Initial best loss value (used when resuming). |
None
|