torch.optim.Optimizer.load_state_dict

Optimizer.load_state_dict(state_dict)

加载优化器状态。

参数

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

注意

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