From 9a48f23726f8cc9bd7c9f318ea1744cb099563c8 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sun, 29 Sep 2019 02:51:24 +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 8b820a97..31532d9f 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -479,7 +479,7 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.5):
 
         # Select predicted classes
         class_conf = class_conf[i]
-        class_pred = class_pred[i].unsqueeze(1).type_as(class_conf)
+        class_pred = class_pred[i].unsqueeze(1).float()
 
         # Box (center x, center y, width, height) to (x1, y1, x2, y2)
         pred[:, :4] = xywh2xyxy(pred[:, :4])