Searched defs:optim_hook (Results 1 – 2 of 2) sorted by relevance
215 def optim_hook(param: torch.nn.Parameter) -> None: function
306 def optim_hook(param: nn.Parameter) -> None: function