Skip to content

Commit 78ce56b

Browse files
committed
code formatting
1 parent d1df8e6 commit 78ce56b

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

monai/networks/blocks/cablock.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
class FeedForward(nn.Module):
2626
"""Gated-DConv Feed-Forward Network (GDFN) that controls feature flow using gating mechanism.
2727
Uses depth-wise convolutions for local context mixing and GELU-activated gating for refined feature selection.
28-
28+
2929
Args:
3030
spatial_dims: Number of spatial dimensions (2D or 3D)
3131
dim: Number of input channels
@@ -78,7 +78,7 @@ class CABlock(nn.Module):
7878
by operating on feature channels instead of spatial dimensions. Incorporates depth-wise
7979
convolutions for local mixing before attention, achieving linear complexity vs quadratic
8080
in vanilla attention. Based on SW Zamir, et al., 2022 <https://arxiv.org/abs/2111.09881>
81-
81+
8282
Args:
8383
spatial_dims: Number of spatial dimensions (2D or 3D)
8484
dim: Number of input channels

monai/networks/nets/restormer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,7 @@ def __init__(self, spatial_dims: int, in_channels: int = 3, embed_dim: int = 48,
8181
conv_only=True,
8282
)
8383

84+
8485
def forward(self, x: torch.Tensor) -> torch.Tensor:
8586
return super().forward(x)
8687

0 commit comments

Comments
 (0)