torch.optim.Optimizer.register_step_post_hook — PyTorch 2.7 documentation (original) (raw)

• Docs >
• torch.optim >
• torch.optim.Optimizer.register_step_post_hook
• 

Shortcuts

Optimizer.register_step_post_hook(hook)[source][source]¶

Register an optimizer step post hook which will be called after optimizer step.

It should have the following signature:

hook(optimizer, args, kwargs) -> None

The optimizer argument is the optimizer instance being used.

Parameters

hook (Callable) – The user defined hook to be registered.

Returns

a handle that can be used to remove the added hook by callinghandle.remove()

Return type

torch.utils.hooks.RemovableHandle

• torch.optim.Optimizer.register_step_post_hook
  • Optimizer.register_step_post_hook()

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.