updates
This commit is contained in:
parent
7b13af707d
commit
82124805f8
|
@ -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()
|
||||
|
|
Loading…
Reference in New Issue