From 7ef7501c362d8180c6432e5dbec7daa8223b1b81 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 27 Dec 2019 12:14:01 -0800 Subject: [PATCH] updates --- utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/utils.py b/utils/utils.py index 7448496c..ba22e6b2 100755 --- 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