Fix docstrings

NKI-AI · Mar 26, 2024 · 1503e50 · 1503e50
1 parent 631e32c
commit 1503e50
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/direct/nn/vsharp/vsharp.py b/direct/nn/vsharp/vsharp.py
@@ -131,9 +131,11 @@ class VSharpNet(nn.Module):
     unrolling a gradient descent scheme (DC-GD). The u-step involves updating the Lagrange multiplier u.
     These steps are iterated for a specified number of cycles.
 
-    The model supports both image and k-space domain parameterizations and includes an initializer for
-    Lagrange multipliers. It also allows for the incorporation of auxiliary steps during training to
-    enhance performance. VSharpNet is tailored for 2D data processing.
+    The model includes an initializer for Lagrange multipliers.
+
+    It also allows for outputting auxiliary steps.
+
+    :class:`VSharpNet` is tailored for 2D MRI data reconstruction.
 
     References
     ----------