Skip to content

Commit

Permalink
code formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
phisanti committed Feb 7, 2025
1 parent d1df8e6 commit 78ce56b
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
4 changes: 2 additions & 2 deletions monai/networks/blocks/cablock.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
class FeedForward(nn.Module):
"""Gated-DConv Feed-Forward Network (GDFN) that controls feature flow using gating mechanism.
Uses depth-wise convolutions for local context mixing and GELU-activated gating for refined feature selection.
Args:
spatial_dims: Number of spatial dimensions (2D or 3D)
dim: Number of input channels
Expand Down Expand Up @@ -78,7 +78,7 @@ class CABlock(nn.Module):
by operating on feature channels instead of spatial dimensions. Incorporates depth-wise
convolutions for local mixing before attention, achieving linear complexity vs quadratic
in vanilla attention. Based on SW Zamir, et al., 2022 <https://arxiv.org/abs/2111.09881>
Args:
spatial_dims: Number of spatial dimensions (2D or 3D)
dim: Number of input channels
Expand Down
1 change: 1 addition & 0 deletions monai/networks/nets/restormer.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,6 +81,7 @@ def __init__(self, spatial_dims: int, in_channels: int = 3, embed_dim: int = 48,
conv_only=True,
)


def forward(self, x: torch.Tensor) -> torch.Tensor:
return super().forward(x)

Expand Down

0 comments on commit 78ce56b

Please sign in to comment.