Skip to content

Commit bbdc0e6

Browse files
committed
del useless code
1 parent f0fa6d9 commit bbdc0e6

1 file changed

Lines changed: 0 additions & 3 deletions

File tree

modules/optimizer/muon.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -122,9 +122,6 @@ def gram_newton_schulz(G: Tensor, steps: int, reset_iterations: List[int]) -> Te
122122
A = torch.bmm(X, X.mT)
123123
B = torch.baddbmm(A, A, A, beta=b_i, alpha=c_i)
124124
X = torch.baddbmm(X, B, X, beta=a_i, alpha=1.0)
125-
126-
if should_transpose:
127-
X = X.mT
128125

129126
return X.to(dtype).view(original_shape)
130127

0 commit comments

Comments
 (0)