yolo.py bug fix

2020-07-13 15:21:19 -07:00 · 2020-07-13 15:21:19 -07:00 · 2b1892430f
parent 02445d176d
commit 2b1892430f
1 changed files with 1 additions and 2 deletions
--- 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):