From e77af92318dc28dd6f9bfb1883f9f7d9e5d370c0 Mon Sep 17 00:00:00 2001
From: Lionel Peer <lionel.peer@gmail.com>
Date: Sun, 8 Oct 2023 15:39:07 +0200
Subject: [PATCH] added losses  to docs

---
 docs/source/api.rst | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/docs/source/api.rst b/docs/source/api.rst
index 9420c80..334f9a0 100644
--- 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