TensorPlay
API symbolsoptim
Copy
View MarkdownDownload .md

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]
Ask DeepWiki