yolo.py bug fix

fhkiel-mlaip · Jul 13, 2020 · 2b18924 · 2b18924
1 parent 02445d1
commit 2b18924
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/models/yolo.py b/models/yolo.py
@@ -74,8 +74,7 @@ def __init__(self, cfg='yolov5s.yaml', ch=3, nc=None):  # model, input channels,
 
         # Init weights, biases
         torch_utils.initialize_weights(self)
-        self._initialize_biases()  # only run once
-        torch_utils.model_info(self)
+        self.info()
         print('')
 
     def forward(self, x, augment=False, profile=False):