MnistModel.state_dict#
- MnistModel.state_dict(destination=None, prefix='', keep_vars=False)#
Returns a dictionary containing a whole state of the module.
Both parameters and persistent buffers (e.g. running averages) are included. Keys are corresponding parameter and buffer names. Parameters and buffers set to
None
are not included.- Returns:
- dict:
a dictionary containing a whole state of the module
Example:
>>> module.state_dict().keys() ['bias', 'weight']