sleepless.utils.checkpointer#
Classes
|
A simple pytorch checkpointer. |
- class sleepless.utils.checkpointer.Checkpointer(model, optimizer=None, scheduler=None, path='.')[source]#
Bases:
object
A simple pytorch checkpointer.
- Parameters:
model – Network model, eventually loaded from a checkpointed file
optimizer – Optimizer
scheduler – Learning rate scheduler
path – Directory where to save checkpoints.
- load(f=None)[source]#
Loads model, optimizer and scheduler from file.
- Parameters:
f (
str
) – Name of a file (absolute or relative toself.path
), that contains the checkpoint data to load into the model, and optionally into the optimizer and the scheduler. If not specified, loads data from current path.