فهرست منبع

modify lr scheduler

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