From bc69220782dedeeac9cbd44f5a3b9390aa745e21 Mon Sep 17 00:00:00 2001 From: Yonghye Kwon Date: Thu, 14 Jan 2021 02:13:13 +0900 Subject: [PATCH] remove unused variable in def compute_loss function (#1659) remove unused variable in def compute_loss function --- utils/loss.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/utils/loss.py b/utils/loss.py index 2cfd0967..844d5039 100644 --- a/utils/loss.py +++ b/utils/loss.py @@ -104,7 +104,6 @@ def compute_loss(p, targets, model): # predictions, targets, model BCEcls, BCEobj = FocalLoss(BCEcls, g), FocalLoss(BCEobj, g) # Losses - nt = 0 # number of targets balance = [4.0, 1.0, 0.4, 0.1] # P3-P6 for i, pi in enumerate(p): # layer index, layer predictions b, a, gj, gi = indices[i] # image, anchor, gridy, gridx @@ -112,7 +111,6 @@ def compute_loss(p, targets, model): # predictions, targets, model n = b.shape[0] # number of targets if n: - nt += n # cumulative targets ps = pi[b, a, gj, gi] # prediction subset corresponding to targets # Regression