TensorPlay
API symbolsoptim
Copy
View MarkdownDownload .md

tensorplay.optim.optimizer.register_optimizer_step_pre_hook

tensorplay.optim.optimizer.register_optimizer_step_pre_hook(hook: Callable[[Optimizer, tuple[Any, ...], dict[str, Any]], tuple[tuple[Any, ...], dict[str, Any]] | None]) RemovableHandle[source]
Ask DeepWiki