update: name

kozistr · kozistr · commit 4fac60dff71c · 2025-03-16T13:28:58.000+09:00
diff --git a/pytorch_optimizer/optimizer/ademamix.py b/pytorch_optimizer/optimizer/ademamix.py
@@ -214,7 +214,7 @@ def __init__(
         super().__init__(params, defaults)
 
     def __str__(self) -> str:
-        return 'SimAdEMAMix'
+        return 'SimplifiedAdEMAMix'
 
     @torch.no_grad()
     def reset(self):
