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 之前，hook 會以參數 self 和 state_dict 被呼叫。註冊的 hook 可用於在呼叫 load_state_dict 之前執行前處理。

參數

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

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

回傳

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

回傳類型

torch.utils.hooks.RemoveableHandle