From 7b74705e98a0651bcad2db1ffc5f6da8cd3ce0f3 Mon Sep 17 00:00:00 2001
From: LJNDDUP <46049065+LJNDDUP@users.noreply.github.com>
Date: Mon, 23 Sep 2019 11:00:11 +0800
Subject: [PATCH] Update utils.py

line 75
labels = labels.float()
---
 utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/utils.py b/utils.py
index a81c3be..35b7ffa 100644
--- a/utils.py
+++ b/utils.py
@@ -72,6 +72,7 @@ def get_depth_sid(args, labels):
     else:
         print('No Dataset named as ', args.dataset)
 
+    labels = labels.float()
     if torch.cuda.is_available():
         alpha_ = torch.tensor(min).cuda()
         beta_ = torch.tensor(max).cuda()