torch.optim.Optimizer.register_step_post_hook

Optimizer.register_step_post_hook(hook)

註冊一個最佳化器步後鉤子，它將在最佳化器步（step）之後被呼叫。

它應該具有以下簽名：

hook(optimizer, args, kwargs) -> None

引數 optimizer 是正在使用的最佳化器例項。

引數

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

返回

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

返回型別

torch.utils.hooks.RemovableHandle