refactor: import

kozistr · Nov 23, 2024 · 4576fb7 · 4576fb7
1 parent e731a85
commit 4576fb7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pytorch_optimizer/__init__.py b/pytorch_optimizer/__init__.py
@@ -45,19 +45,22 @@
     BSAM,
     CAME,
     FTRL,
+    GSAM,
     LARS,
     LOMO,
     MADGRAD,
     MSVAG,
     PID,
     PNM,
     QHM,
+    SAM,
     SGDP,
     SGDW,
     SM3,
     SOAP,
     SRMM,
     SWATS,
+    WSAM,
     A2Grad,
     AccSGD,
     AdaBelief,