torch.optim.Optimizer.register_step_post_hook

Optimizer.register_step_post_hook(hook)

注册优化器步骤后钩子，该钩子在优化器步骤之后将被调用。

它应该具有以下签名：

hook(optimizer, args, kwargs) -> None

第 0#个参数是正在使用的优化器实例。

参数:

hook（可调用对象）- 用户定义的将被注册的钩子。

返回值:

一个可以用来通过调用 handle.remove() 移除已添加钩子的句柄。

返回类型:

torch.utils.hooks.RemovableHandle