updates

2018-11-22 17:17:07 +01:00 · 2018-11-22 17:17:07 +01:00 · 7ca924b172
commit 7ca924b172
parent 075b629049
1 changed files with 1 additions and 0 deletions
--- a/models.py
+++ b/models.py
@ -184,6 +184,7 @@ class YOLOLayer(nn.Module):

            # Sum loss components
            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()