未验证 提交 7282f554 编写于 作者: F Francisco Massa 提交者: GitHub

Fix weight init in FastRCNNPredictor (#60)

上级 3308cd92
......@@ -17,10 +17,10 @@ class FastRCNNPredictor(nn.Module):
self.bbox_pred = nn.Linear(num_inputs, num_classes * 4)
nn.init.normal_(self.cls_score.weight, mean=0, std=0.01)
nn.init.constant_(self.cls_score.weight, 0)
nn.init.constant_(self.cls_score.bias, 0)
nn.init.normal_(self.bbox_pred.weight, mean=0, std=0.001)
nn.init.constant_(self.bbox_pred.weight, 0)
nn.init.constant_(self.bbox_pred.bias, 0)
def forward(self, x):
x = self.avgpool(x)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册