added losses to docs

liopeer · Oct 8, 2023 · e77af92 · e77af92
1 parent b4e9dc4
commit e77af92
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/docs/source/api.rst b/docs/source/api.rst
@@ -15,6 +15,13 @@ API
 
     diffusion_models.mri_forward
 
+.. autosummary::
+    :toctree: _autosummary
+    :template: custom-module-template.rst
+    :recursive:
+
+    diffusion_models.losses
+
 .. autosummary::
     :toctree: _autosummary
     :template: custom-module-template.rst