π [Fix] a bug in SPPELAN module
Browse files- yolo/model/module.py +1 -1
yolo/model/module.py
CHANGED
@@ -80,7 +80,7 @@ class SPPELAN(nn.Module):
|
|
80 |
neck_channels = neck_channels or out_channels // 2
|
81 |
|
82 |
self.conv1 = Conv(in_channels, neck_channels, kernel_size=1)
|
83 |
-
self.pools = nn.ModuleList([Pool("max", 5, padding=
|
84 |
self.conv5 = Conv(4 * neck_channels, out_channels, kernel_size=1)
|
85 |
|
86 |
def forward(self, x: Tensor) -> Tensor:
|
|
|
80 |
neck_channels = neck_channels or out_channels // 2
|
81 |
|
82 |
self.conv1 = Conv(in_channels, neck_channels, kernel_size=1)
|
83 |
+
self.pools = nn.ModuleList([Pool("max", 5, padding=2, stride=1) for _ in range(3)])
|
84 |
self.conv5 = Conv(4 * neck_channels, out_channels, kernel_size=1)
|
85 |
|
86 |
def forward(self, x: Tensor) -> Tensor:
|