⚗️ [Add] a experiment, a bias init formula
Browse files- 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)
|