state_dict — упорядоченный словарь имён → тензоры параметров и буферов Module/Optimizer.
Разбор
- Ключи отражают иерархию
layer.weight.
- У Optimizer свой state_dict (моменты Adam).
- Чекпоинт часто включает model+optim+epoch+scaler.
- Полезен для partial load и отладки missing keys.
Итог
state_dict — сериализуемый снимок состояния обучения.