Update docs

lululxvi · Jan 4, 2025 · 6a06766 · 6a06766
1 parent a3c677c
commit 6a06766
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 15 deletions.
diff --git a/docs/index.rst b/docs/index.rst
@@ -107,6 +107,7 @@ If you are looking for information on a specific function, class or method, this
   modules/deepxde.nn.tensorflow
   modules/deepxde.nn.tensorflow_compat_v1
   modules/deepxde.optimizers
+  modules/deepxde.optimizers.pytorch
   modules/deepxde.utils
 
 Indices and tables

diff --git a/docs/modules/deepxde.optimizers.pytorch.rst b/docs/modules/deepxde.optimizers.pytorch.rst
@@ -1,21 +1,11 @@
-deepxde.optimizers.pytorch package
-==================================
+deepxde.optimizers.pytorch
+==========================
 
-Submodules
-----------
+deepxde.optimizers.pytorch.nncg module
+--------------------------------------
 
-deepxde.optimizers.pytorch.optimizers module
---------------------------------------------
-
-.. automodule:: deepxde.optimizers.pytorch.optimizers
+.. automodule:: deepxde.optimizers.pytorch.nncg
    :members:
    :undoc-members:
    :show-inheritance:
 
-Module contents
----------------
-
-.. automodule:: deepxde.optimizers.pytorch
-   :members:
-   :undoc-members:
-   :show-inheritance: