Searched defs:unscale_ (Results 1 – 2 of 2) sorted by relevance
264 def unscale_(self, optimizer: torch.optim.Optimizer) -> None: member in ShardedGradScaler
287 def unscale_(self, optimizer: torch.optim.Optimizer) -> None: member in GradScaler