torch.optim.Optimizer.register_step_pre_hook#
- Optimizer.register_step_pre_hook(hook)[source]#
註冊一個最佳化器步驟預鉤子，它將在最佳化器步驟之前被呼叫。
它應具有以下簽名
hook(optimizer, args, kwargs) -> None or modified args and kwargs
引數
optimizer是正在使用的最佳化器例項。如果 args 和 kwargs 被前置鉤子修改，則轉換後的值將作為包含 new_args 和 new_kwargs 的元組返回。- 引數
hook (Callable) – 使用者定義的待註冊鉤子。
- 返回
一個控制代碼，可用於透過呼叫
handle.remove()來移除新增的鉤子- 返回型別
torch.utils.hooks.RemovableHandle
