yjh0410 пре 1 година
родитељ
комит
807860f342
1 измењених фајлова са 0 додато и 3 уклоњено
  1. 0 3
      models/yolov3/yolov3_pred.py

+ 0 - 3
models/yolov3/yolov3_pred.py

@@ -45,9 +45,6 @@ class DetPredLayer(nn.Module):
         b = self.reg_pred.bias.view(-1, )
         b.data.fill_(1.0)
         self.reg_pred.bias = torch.nn.Parameter(b.view(-1), requires_grad=True)
-        w = self.reg_pred.weight
-        w.data.fill_(0.)
-        self.reg_pred.weight = torch.nn.Parameter(w, requires_grad=True)
 
     def generate_anchors(self, fmp_size):
         """