Przeglądaj źródła

add Albu & modify head init

yjh0410 2 lat temu
rodzic
commit
2cbac1ef6d
1 zmienionych plików z 3 dodań i 3 usunięć
  1. 3 3
      models/detectors/yolov8/yolov8_pred.py

+ 3 - 3
models/detectors/yolov8/yolov8_pred.py

@@ -30,9 +30,9 @@ class SingleLevelPredLayer(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)
+        # w = self.reg_pred.weight
+        # w.data.fill_(0.)
+        # self.reg_pred.weight = torch.nn.Parameter(w, requires_grad=True)
 
     def forward(self, cls_feat, reg_feat):
         """