torch.optim.Optimizer.register_state_dict_post_hook

Optimizer.register_state_dict_post_hook(hook, prepend=False)

註冊一個狀態字典後置鉤子，該鉤子將在呼叫 state_dict() 後被呼叫。

它應具有以下簽名：

hook(optimizer, state_dict) -> state_dict or None

鉤子將在為 self 生成 state_dict 後，以引數 self 和 state_dict 呼叫。鉤子可以原地修改 state_dict，或者選擇返回一個新的 state_dict。註冊的鉤子可用於在 state_dict 返回之前對其進行後處理。

引數

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

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

返回

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

返回型別

torch.utils.hooks.RemoveableHandle