diff --git a/models/common.py b/models/common.py index db2b89de2..f57dc32d8 100644 --- a/models/common.py +++ b/models/common.py @@ -1048,7 +1048,8 @@ class GhostBottleneck(nn.Module): ) def forward(self, x): - """```"""Processes input tensor through sequential GhostNet convolution layers and optional shortcut. + """ + Processes input tensor through sequential GhostNet convolution layers and optional shortcut. Args: x (torch.Tensor): Input tensor with shape (N, C, H, W) where N is the batch size, C is the number of channels, @@ -1073,8 +1074,6 @@ class GhostBottleneck(nn.Module): The function harnesses the GhostNet architecture, enabling efficient feature extraction with reduced computational complexity, as detailed in https://github.com/huawei-noah/ghostnet. """ - ``` - """ return self.conv(x) + self.shortcut(x)