diff --git a/models.py b/models.py
index fd404375..1b12bd94 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 = False
+            balance_losses_flag = True
             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()