oneflow.nn.Graph.state_dict

Graph.state_dict(destination=None)Dict[str, Union[Dict[str, oneflow.Tensor], oneflow.Tensor]]

Returns a dictionary containing a whole state of the graph.

States of modules/optimizers/lr schedulers in a graph are included.

Keys of modules’ state dict are corresponding to their name in the graph. Values of modules’ state dict are corresponding to their nn.Module’s state dict.

Other keys and tensors are states of optimizers/lr schedulers/etc.

Returns

a dictionary containing the whole state of the graph.

Return type

dict