Skip to content

Commit 5f17f7a

Browse files
committed
run black with proper args
1 parent 83204c0 commit 5f17f7a

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

pytorch_optimizer/optimizer/orthograd.py

+2-6
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,7 @@ def __init__(self, optimizer: OPTIMIZER_INSTANCE_OR_CLASS, **kwargs) -> None:
3535
params = kwargs.pop("params")
3636
self.optimizer = optimizer(params, **kwargs)
3737
else:
38-
raise ValueError(
39-
"Need to pass `params` when you pass the torch.optim.Optimizer instance."
40-
)
38+
raise ValueError("Need to pass `params` when you pass the torch.optim.Optimizer instance.")
4139

4240
self.defaults: DEFAULTS = self.optimizer.defaults
4341

@@ -66,9 +64,7 @@ def orthogonalize_gradients(self, params) -> None:
6664

6765
proj = torch.dot(w, g).div_(torch.dot(w, w).add_(self.eps))
6866
g_ortho = g.to(dtype=torch.float32, copy=True).sub_(w, alpha=proj)
69-
g_ortho_scaled = g_ortho.mul_(
70-
g.norm(2).div_(g_ortho.norm(2).add_(self.eps))
71-
)
67+
g_ortho_scaled = g_ortho.mul_(g.norm(2).div_(g_ortho.norm(2).add_(self.eps)))
7268

7369
p.grad.copy_(g_ortho_scaled.view_as(p.grad))
7470

0 commit comments

Comments
 (0)