Update common.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>pull/13194/head
parent
49728e4002
commit
df7b908cc8
|
@ -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)
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue