[skip ci] docs: running_stats

kozistr · kozistr · commit 99955925062c · 2023-01-24T21:55:43.000+09:00
diff --git a/docs/util_api.rst b/docs/util_api.rst
@@ -56,3 +56,20 @@ SafeFP16Optimizer
 
 .. autoclass:: pytorch_optimizer.SafeFP16Optimizer
     :members:
+
+.. _enable_running_stats:
+
+enable_running_stats
+--------------------
+
+.. autoclass:: pytorch_optimizer.enable_running_stats
+    :members:
+
+
+.. _disable_running_stats:
+
+disable_running_stats
+---------------------
+
+.. autoclass:: pytorch_optimizer.disable_running_stats
+    :members:
