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_names ，则将保存并覆盖优化器状态中当前存在的名称。如果它们不在加载的状态字典中，则优化器的 param_names 将保持不变。