torch.optim.Optimizer.register_step_pre_hook

Optimizer.register_step_pre_hook(hook)

註冊一個 optimizer step 前置鉤子，它將在 optimizer step 之前被呼叫。

它應該具有以下簽名

hook(optimizer, args, kwargs) -> None or modified args and kwargs

optimizer 引數是正在使用的 optimizer 例項。如果 args 和 kwargs 被前置鉤子修改，則修改後的值將作為包含 new_args 和 new_kwargs 的元組返回。

引數

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

返回

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

返回型別

torch.utils.hooks.RemovableHandle