Searched defs:OptimizerPreHook (Results 1 – 1 of 1) sorted by relevance
333 …OptimizerPreHook: TypeAlias = Callable[[Self, Args, Kwargs], Optional[Tuple[Args, Kwargs]]] # typ… variable in Optimizer527 def register_step_pre_hook(self, hook: OptimizerPreHook) -> RemovableHandle: