henry000 commited on
Commit
31c2c1a
Β·
1 Parent(s): d8f15ea

πŸ› [Fix] a bug in SPPELAN module

Browse files
Files changed (1) hide show
  1. 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=0) for _ in range(3)])
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: