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