From 259e43ede8361b56e00bd3b8d3286be0e154df17 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 2 Feb 2021 17:59:16 +0100
Subject: [PATCH] Cannot only use Float16, need Float32.
---
pysegcnn/core/dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pysegcnn/core/dataset.py b/pysegcnn/core/dataset.py
index 00b3fda..ba9df96 100644
--- a/pysegcnn/core/dataset.py
+++ b/pysegcnn/core/dataset.py
@@ -346,7 +346,7 @@ class ImageDataset(Dataset):
x, y = scene['transform'](x, y)
# convert to torch tensors
- x = self.to_tensor(x, dtype=torch.float16)
+ x = self.to_tensor(x, dtype=torch.float32)
y = self.to_tensor(y, dtype=torch.uint8)
return x, y
--
GitLab