henry000 commited on
Commit
425a4ef
Β·
1 Parent(s): 92614f8

πŸ› [Update] Conv to default no bias

Browse files
Files changed (1) hide show
  1. yolo/model/module.py +1 -1
yolo/model/module.py CHANGED
@@ -24,7 +24,7 @@ class Conv(nn.Module):
24
  ):
25
  super().__init__()
26
  kwargs.setdefault("padding", auto_pad(kernel_size, **kwargs))
27
- self.conv = nn.Conv2d(in_channels, out_channels, kernel_size, **kwargs)
28
  self.bn = nn.BatchNorm2d(out_channels)
29
  self.act = get_activation(activation)
30
 
 
24
  ):
25
  super().__init__()
26
  kwargs.setdefault("padding", auto_pad(kernel_size, **kwargs))
27
+ self.conv = nn.Conv2d(in_channels, out_channels, kernel_size, bias=False, **kwargs)
28
  self.bn = nn.BatchNorm2d(out_channels)
29
  self.act = get_activation(activation)
30