From 82124805f8a6fd5b44e4287649bed9abf61ca021 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 23 Nov 2018 18:13:35 +0100 Subject: [PATCH] updates --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 1b12bd94..fd404375 100755 --- a/models.py +++ b/models.py @@ -183,7 +183,7 @@ class YOLOLayer(nn.Module): # lconf += k * BCEWithLogitsLoss(pred_conf[~mask], mask[~mask].float()) # Sum loss components - balance_losses_flag = True + balance_losses_flag = False if balance_losses_flag: k = 1 / self.loss_means.clone() loss = (lx * k[0] + ly * k[1] + lw * k[2] + lh * k[3] + lconf * k[4] + lcls * k[5]) / k.mean()