torch.optim.Optimizer.register_load_state_dict_pre_hook

Optimizer.register_load_state_dict_pre_hook(hook, prepend=False)

註冊一個 load_state_dict 預鉤子，該鉤子將在呼叫 load_state_dict() 之前被呼叫。它應具有以下簽名

hook(optimizer, state_dict) -> state_dict or None

引數 optimizer 是正在使用的最佳化器例項，引數 state_dict 是使用者傳遞給 load_state_dict 的 state_dict 的淺複製。該鉤子可以就地修改 state_dict，或者選擇返回一個新的 state_dict。如果返回了一個 state_dict，它將被用來載入到最佳化器中。

在對 self 呼叫 load_state_dict 之前，將使用引數 self 和 state_dict 呼叫該鉤子。註冊的鉤子可用於在呼叫 load_state_dict 之前執行預處理。

引數

• hook (Callable) – 要註冊的使用者定義鉤子。

• prepend (bool) – 如果為 True，提供的預鉤子 hook 將在所有已註冊的 load_state_dict 預鉤子之前觸發。否則，提供的鉤子 hook 將在所有已註冊的預鉤子之後觸發。(預設值: False)

返回值

一個控制代碼，可以透過呼叫 handle.remove() 來移除新增的鉤子

返回型別

torch.utils.hooks.RemoveableHandle