diff --git a/utils/utils.py b/utils/utils.py
index 5b4e9c76..e0041069 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -278,7 +278,8 @@ def compute_loss(p, targets):  # predictions, targets
             tconf[b, a, gj, gi] = 1  # conf
 
             lxy += (k * 8) * MSE(torch.sigmoid(pi[..., 0:2]), txy[i])  # xy loss
-            lwh += (k * 4) * MSE(pi[..., 2:4], twh[i])  # wh loss
+            lwh += (k * 4) * MSE(torch.sigmoid(pi[..., 2:4]), twh[i])  # wh yolo loss
+            # lwh += (k * 4) * MSE(torch.sigmoid(pi[..., 2:4]), twh[i])  # wh power loss
             lcls += (k * 1) * CE(pi[..., 5:], tcls[i])  # class_conf loss
 
         # pos_weight = FT([gp[i] / min(gp) * 4.])