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