From ebd93e354d4ae75bfa7cc163bb3ae615ac284641 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sun, 18 Aug 2019 21:28:49 +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 ef6c61e8..e0e87bc2 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -327,7 +327,7 @@ def compute_loss(p, targets, model):  # predictions, targets, model
     bs = p[0].shape[0]  # batch size
     k = bs / 64  # loss gain
     arc = 'normal'  # (normal, uCE, uBCE, uBCEs) detection architectures
-    for i, pi0 in enumerate(p):  # layer i predictions, i
+    for i, pi in enumerate(p):  # layer index, layer predictions
         b, a, gj, gi = indices[i]  # image, anchor, gridy, gridx
         tobj = torch.zeros_like(pi[..., 0])  # target obj