oneflow.optim.Optimizer.state_dict¶
-
Optimizer.
state_dict
()¶ Returns the state of the optimizer as a
dict
.It contains two entries:
state - a dict holding current optimization state. Its content differs between optimizer classes.
param_group - a dict containing all parameter groups.
It almost copied from: https://pytorch.org/docs/1.10/_modules/torch/optim/optimizer.html#Optimizer.state_dict.