Glossary: Deep Learning Basics
state_dict
Also written: state dict
A dictionary mapping each parameter's name to its tensor, for a model or an optimizer. It's what you save with `torch.save` and restore with `load_state_dict`.
Glossary: Deep Learning Basics
Also written: state dict
A dictionary mapping each parameter's name to its tensor, for a model or an optimizer. It's what you save with `torch.save` and restore with `load_state_dict`.