Spaces:

xiang-wuu
/

yolov5

Runtime error

glenn-jocher commited on Jul 13, 2020

Commit

2b18924

1 Parent(s): 02445d1

yolo.py bug fix

Files changed (1) hide show

models/yolo.py CHANGED Viewed

@@ -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)
         print('')
     def forward(self, x, augment=False, profile=False):

         # Init weights, biases
         torch_utils.initialize_weights(self)
+        self.info()
         print('')
     def forward(self, x, augment=False, profile=False):