Update common.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
pull/13194/head
Glenn Jocher 2024-07-17 17:32:23 +02:00 committed by GitHub
parent 49728e4002
commit df7b908cc8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions

View File

@ -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)