Update monai/networks/blocks/spade_norm.py

Co-authored-by: YunLiu <55491388+KumoLiu@users.noreply.github.com> Signed-off-by: Mark Graham <markgraham539@gmail.com>
marksgraham · Dec 18, 2023 · 51f72e2 · 51f72e2
1 parent 035d2d2
commit 51f72e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/monai/networks/blocks/spade_norm.py b/monai/networks/blocks/spade_norm.py
@@ -81,7 +81,7 @@ def forward(self, x: torch.Tensor, segmap: torch.Tensor) -> torch.Tensor:
         """
         Args:
             x: input tensor
-            segmap: input segmentation map (bxcx[spatial-dimensions]) where c is the number of semantic channels.
+            segmap: input segmentation map (B, C, [spatial-dimensions]) where C is the number of semantic channels.
             The map will be interpolated to the dimension of x internally.
         """