The python torch.optim.SGD.state_dict function returns a dictionary containing a state of the SGD optimizer. This dictionary includes current values of parameters, buffers, and additional states for the optimizer. The state can be saved for future use or for restoring the optimizer's state at a later time.
Python SGD.state_dict - 30 examples found. These are the top rated real world Python examples of torch.optim.SGD.state_dict extracted from open source projects. You can rate examples to help us improve the quality of examples.