Docs4dev
Docs
pytorch / 2 / generated / torch.optim.optimizer.load_state_dict.html

torch.optim.Optimizer.load_state_dict

Optimizer.load_state_dict(state_dict) [source]

Loads the optimizer state.

Parameters

state_dict (dict) – optimizer state. Should be an object returned from a call to state_dict().

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.optim.Optimizer.load_state_dict.html