diff --git a/models/yolo.py b/models/yolo.py index 3a34a28..da96a31 100644 --- a/models/yolo.py +++ b/models/yolo.py @@ -74,8 +74,7 @@ class Model(nn.Module): # 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):