From ccfd44c2f87d60a5281372fc2c32318262bb1e62 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 27 Apr 2019 18:36:19 +0200
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 59fe1a8c..2fb8c75d 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -259,7 +259,7 @@ def compute_loss(p, targets, model):  # predictions, targets, model
 
     # Define criteria
     MSE = nn.MSELoss()
-    CE = nn.CrossEntropyLoss(weight=model.class_weights)
+    CE = nn.CrossEntropyLoss()  # (weight=model.class_weights)
     BCE = nn.BCEWithLogitsLoss()
 
     # Compute losses