From 7046a95d61609e156f5162982b333859883a4611 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 21 Aug 2019 00:23:41 +0200
Subject: [PATCH] updates

---
 utils/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/utils.py b/utils/utils.py
index fc575267..2454485c 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -367,7 +367,7 @@ def compute_loss(p, targets, model):  # predictions, targets, model
             t = torch.zeros_like(pi[..., 5:])  # targets
             if nb:
                 t[b, a, gj, gi, tcls[i]] = 1.0
-            lcls += BCEobj(pi[..., 5:], t)
+            lobj += BCEobj(pi[..., 5:], t)
 
     lbox *= k * h['giou']
     lobj *= k * h['obj']