diff --git a/models/yolo.py b/models/yolo.py index 3a34a280f669..da96a3102084 100644 --- 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):