state_dict

Also written: load_state_dict

A PyTorch dictionary of a model's weights and buffers. `state_dict()` saves them and `load_state_dict` restores them into a model of the same sizes.