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):
         """