docs: WSAM optimizer

kozistr · Dec 10, 2023 · 1b4a393 · 1b4a393
1 parent 5ea4bec
commit 1b4a393
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/docs/optimizer.md b/docs/optimizer.md
@@ -272,6 +272,10 @@
     :docstring:
     :members:
 
+::: pytorch_optimizer.WSAM
+    :docstring:
+    :members:
+
 ::: pytorch_optimizer.Yogi
     :docstring:
     :members: