Update docs with new optimizers. (#90)

jettify · web-flow · commit 5155f0b3c206 · 2020-03-21T21:31:43.000-04:00
diff --git a/docs/api.rst b/docs/api.rst
@@ -32,6 +32,36 @@ Lamb
 .. autoclass:: torch_optimizer.Lamb
     :members:
 
+NovoGrad
+--------
+
+.. autoclass:: torch_optimizer.NovoGrad
+    :members:
+
+PID
+---
+
+.. autoclass:: torch_optimizer.PID
+    :members:
+
+QHAdam
+------
+
+.. autoclass:: torch_optimizer.QHAdam
+    :members:
+
+QHM
+---
+
+.. autoclass:: torch_optimizer.QHM
+    :members:
+
+RAdam
+-----
+
+.. autoclass:: torch_optimizer.RAdam
+    :members:
+
 SGDW
 ----
 
