torch.optim.Optimizer.load_state_dict

Optimizer.load_state_dict(state_dict)

加载优化器状态。

参数

state_dict (dict) – 优化器状态。应该是由调用 state_dict() 返回的对象。

注意

参数的名称（如果它们存在于 state_dict() 中每个参数组的 “param_names” 键下）不会影响加载过程。对于自定义情况（例如加载的状态字典中的参数与优化器中初始化的参数不同时），应实现自定义的 register_load_state_dict_pre_hook 来相应地调整加载的字典，以便使用参数的名称。如果加载的状态字典的 param_groups 中存在 param_names，它们将被保存并覆盖优化器状态中当前的名称（如果存在）。如果加载的状态字典中不存在它们，则优化器的 param_names 将保持不变。