From 76abdb22d15669e1dab5365aa74a0d9d02b8dd37 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 2 Feb 2021 17:47:29 +0100
Subject: [PATCH] Use Float16 to save memory.
---
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 ba9df96..00b3fda 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.float32)
+ x = self.to_tensor(x, dtype=torch.float16)
y = self.to_tensor(y, dtype=torch.uint8)
return x, y
--
GitLab