torch.optim.Optimizer.register_load_state_dict_post_hook

Optimizer.register_load_state_dict_post_hook(hook, prepend=False)

註冊一個 load_state_dict 後置掛鉤，它將在 load_state_dict() 被調用後調用。它應該具有以下簽名

hook(optimizer) -> None

optimizer 參數是正在使用的優化器實例。

在 self 上調用 load_state_dict 之後，將使用參數 self 調用該掛鉤。註冊的掛鉤可用於在 load_state_dict 載入 state_dict 之後執行後處理。

參數

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

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

回傳

一個控制代碼 (handle)，可用於透過呼叫 handle.remove() 來移除已新增的 hook

回傳類型

torch.utils.hooks.RemoveableHandle