API symbolsoptimCopyCopy as MarkdownCopy page linkView MarkdownDownload .mdOpenAIOpen in ChatGPTClaudeOpen in ClaudePerplexityOpen in PerplexityGrokOpen in Grok tensorplay.optim.optimizer.register_optimizer_step_post_hook tensorplay.optim.optimizer.register_optimizer_step_post_hook(hook: Callable[[Optimizer, tuple[Any, ...], dict[str, Any]], None]) → RemovableHandle[source] Help improve this pageFound an error, an unclear step, or a missing example?Was this page helpful?YesNoSuggest an editReport an issueSWALRAnneal each optimizer learning rate to a fixed SWA learning rate.tensorplay.optim.optimizer.register_optimizer_step_pre_hookNext Page