From 2b1892430f2564df7620f3ee94443ea18c44ed07 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 13 Jul 2020 15:21:19 -0700
Subject: [PATCH] yolo.py bug fix
---
models/yolo.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/models/yolo.py b/models/yolo.py
index 3a34a280f..da96a3102 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):