torch.optim.Optimizer.load_state_dict
Optimizer.load_state_dict(state_dict)
Loads the optimizer state.
Parameters:
      state_dict (dict) -- optimizer state. Should be an object
      returned from a call to "state_dict()".