Update networks.py

Deep-MI · Jun 7, 2024 · 1a38900 · 1a38900
1 parent 016a5ba
commit 1a38900
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/FastSurferCNN/models/networks.py b/FastSurferCNN/models/networks.py
@@ -338,10 +338,9 @@ def forward(
         ----------
         x : Tensor
             Input image [N, C, H, W].
-        scale_factor : Tensor
-            [MISSING] [N, 1].
+            [N, 1] Defaults to None.
         scale_factor_out : Tensor, optional
-            [MISSING].
+            Tensor representing the scale factor for the output. Defaults to None.
 
         Returns
         -------