We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5f530f6 commit 6c752deCopy full SHA for 6c752de
pytorch_optimizer/optimizer/shampoo.py
@@ -37,7 +37,7 @@ def add_statistics(self, grad: torch.Tensor):
37
def precondition_gradient(self, grad: torch.Tensor) -> torch.Tensor:
38
return grad
39
40
- def update_momentum(self, update: torch.Tensor, unused_beta1: float) -> torch.Tensor:
+ def update_momentum(self, update: torch.Tensor, unused_beta1: float) -> torch.Tensor: # noqa: ARG002
41
return update
42
43
0 commit comments