Sfoglia il codice sorgente

modify lr scheduler

yjh0410 1 anno fa
parent
commit
807860f342
1 ha cambiato i file con 0 aggiunte e 3 eliminazioni
  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):
         """