henry000 commited on
Commit
592f75d
·
1 Parent(s): f1585d3

⚗️ [Add] a experiment, a bias init formula

Browse files
Files changed (1) hide show
  1. yolo/model/module.py +1 -1
yolo/model/module.py CHANGED
@@ -81,7 +81,7 @@ class Detection(nn.Module):
81
  self.anc2vec = Anchor2Vec(reg_max=reg_max)
82
 
83
  self.anchor_conv[-1].bias.data.fill_(1.0)
84
- self.class_conv[-1].bias.data.fill_(-10)
85
 
86
  def forward(self, x: Tensor) -> Tuple[Tensor]:
87
  anchor_x = self.anchor_conv(x)
 
81
  self.anc2vec = Anchor2Vec(reg_max=reg_max)
82
 
83
  self.anchor_conv[-1].bias.data.fill_(1.0)
84
+ self.class_conv[-1].bias.data.fill_(-10) # TODO: math.log(5 * 4 ** idx / 80 ** 3)
85
 
86
  def forward(self, x: Tensor) -> Tuple[Tensor]:
87
  anchor_x = self.anchor_conv(x)