Jelajahi Sumber

modify lr scheduler

yjh0410 1 tahun lalu
induk
melakukan
807860f342
1 mengubah file dengan 0 tambahan dan 3 penghapusan
  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):
         """