updates

2019-12-27 12:14:01 -08:00 · 2019-12-27 12:14:01 -08:00 · 7ef7501c36
parent 59de209ab2
commit 7ef7501c36
1 changed files with 1 additions and 1 deletions
--- a/utils/utils.py
+++ b/utils/utils.py
@ -436,7 +436,7 @@ def compute_loss(p, targets, model):  # predictions, targets, model
    lcls *= h['cls']
    if red == 'sum':
        lbox *= 3 / ng
-        lobj *= 3 / np
+        lobj *= 3 / np * 2
        lcls *= 3 / ng / model.nc

    loss = lbox + lobj + lcls