torch.optim.Optimizer.register_step_post_hook

Optimizer.register_step_post_hook(hook)

註冊一個最佳化器步驟後掛鉤，它將在最佳化器步驟之後被呼叫。

它應該具有以下簽章

hook(optimizer, args, kwargs) -> None

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

參數

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

傳回

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

傳回類型

torch.utils.hooks.RemovableHandle