Save & Load with state_dict
Checkpoint weights to resume later.
Why Checkpoints Matter
Training can take hours, and crashes happen. Saving your progress as a checkpoint lets you stop, resume, or ship the model without retraining from zero.
What a state_dict Holds
A model's state_dict is a plain dictionary mapping each layer name to its learned tensors. It is everything the model knows, packed for storage.
model.state_dict()All lessons in this course
- Split Train, Validation & Test
- An Epoch Loop with Validation
- Save & Load with state_dict
- Early Stopping on Val Loss