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